-
pytorch-revnet:在PyTorch中实现可逆残差网络的MATLAB代码
资源介绍
ist的matlab代码转网
实施
。
要求
主要要求显然。
强烈建议使用CUDA。
训练脚本需要进度条。
单元测试需要由PyTorch项目实现的TestCase。
该模块可以下载。
笔记
该项目中的revnet模型往往具有爆炸梯度。
为了解决这个问题，我使用了梯度范数裁剪。
对于以下实验，您可以调用以下命令：
python
train_cifar.py
--model
revnet38
--clip
0.25
结果
CIFAR-10
模型
准确性
内存使用情况
参量
resnet32
92.02％
1271兆字节
47万
revnet38
91.98％
660兆字节
47万