HomebrewNLP/revlib
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
PythonBSD-2-Clause
Issues
- 8
revlib can not work in torch amp.
#7 opened by JAYatBUAA - 1
convolution network, striding
#6 opened by klae01 - 5
- 4
TorchSummary for revlib
#4 opened by taokz - 7
Compute input from output?
#2 opened by isaacrob