Fix CIFAR data loaders - Githubissues

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)

[x] Test passed
[x] Coding style (indentation, etc)

Additional Comments (if any)

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