🐛 fix(nn/utils): `net` doesn't have to be on CPU anymore (now uses device)

Vermeille / Torchelie

Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.

https://torchelie.readthedocs.org/

MIT License

111 stars 11 forks source link

Closed Horgix closed 3 years ago