silvandeleemput / memcnn

PyTorch Framework for Developing Memory Efficient Deep Invertible Networks
MIT License
251 stars 26 forks source link

Fix CIFAR data loaders #76

Closed silvandeleemput closed 1 year ago

silvandeleemput commented 1 year ago

Closes #74

What was the problem?

The CIFAR data loaders could not be pickled under Windows because of the local Lambda operations

How this PR fixes the problem?

The lambda operations have been made explicit as functions

Check lists (check x in [ ] of list items)

Additional Comments (if any)

Needed to update the CI as well, and disabled PyTorch 1.0, 1.1, latest