Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
take
torch.nn.Module
model into account when moving to device (#3167)
* bug fix * update code
- Loading branch information