search

eisen-ai / eisen-core

Core functionality of Eisen
MIT License
41 stars 10 forks source link

Native AMP enabled to reduce the memory footprint. #35

Closed canerozer closed 4 years ago

canerozer commented 4 years ago

As mentioned in #34, PyTorch 1.6.0 is now natively supporting AMP, and hence, the relevant lines have been uncommented.