search

ucbrise / actnn

ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
MIT License
196 stars 30 forks source link

A kind suggestion on the pytorch version. #25

Closed guanchuwang closed 2 years ago

guanchuwang commented 2 years ago

This is a kind suggestion to the authors to revise the required pytorch version to "torch >= 1.7.1" and "torch <= 1.8.0" in the readme file.

merrymercy commented 2 years ago

Thanks for the suggestion. It is updated https://github.com/ucbrise/actnn/commit/370bc9fe5cd8a64d817d5cc3924c6d3a2051db92

You can also find a possible fix in this PR https://github.com/ucbrise/actnn/pull/22