search

tristandeleu / pytorch-meta

A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch
https://tristandeleu.github.io/pytorch-meta/
MIT License
1.98k stars 256 forks source link

Will this package work with pytorch 1.6? #82

Closed lidh15 closed 4 years ago

lidh15 commented 4 years ago

Currently, it cannot be installed with pytorch 1.6, will you update the package, or we have to roll back to 1.4?

tristandeleu commented 4 years ago

Almost all the tests are passing with PyTorch 1.6. The only test failing is DataParallel with params=None (the default behavior), which broke due to a recent change to PyTorch's DataParallel. I will look into it. If you are not using DataParallel, then you should expect it to work with PyTorch 1.6.