Fix bug in compute_tensor_bytes

ucbrise / actnn

ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training

MIT License

196 stars 30 forks source link

Closed Ying1123 closed 1 year ago