Remove Python 2.7 and add PyTorch 1.7 to CI tests

silvandeleemput / memcnn

PyTorch Framework for Developing Memory Efficient Deep Invertible Networks

MIT License

251 stars 26 forks source link

Remove Python 2.7 and add PyTorch 1.7 to CI tests #58

Closed silvandeleemput closed 3 years ago

silvandeleemput commented 3 years ago

Description

Support for Python 2 has been abandoned for a while now and so I will no longer support the Python 2 tests. This should also simplify dealing with issues like #18. Also, new tests for PyTorch 1.6 should be added also to test using the now build-in amp support (torch.cuda.amp). (see #57)