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

Converting batch_meta_collate to a class to make it pickle-able for '… #39

Closed tesfaldet closed 4 years ago

tesfaldet commented 4 years ago

…spawn' method multiprocessing.

tesfaldet commented 4 years ago

For dealing with issue https://github.com/tristandeleu/pytorch-meta/issues/38. This is just a really quick fix that's been loosely tested. Perhaps there's a better way of doing this?