ucbrise / actnn

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

fix import bug on torch 1.9.0 #22

Open liuyanyi opened 3 years ago

liuyanyi commented 3 years ago

fix bug in torch._six

15