ucbrise / actnn

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

Fix a non-contiguous bug & add warnings for pooling layers #3

Closed merrymercy closed 3 years ago