Closed Hannibal046 closed 1 year ago
make c be on the same device as unit, because the default device of torch.tensor is cpu
c
unit
torch.tensor
cpu
This issue is already solved by other pull requests and waiting to be released. Please see develop branch.
slved by #577
make
c
be on the same device asunit
, because the default device oftorch.tensor
iscpu