ist的matlab代码-pytorch-revnet:pytorch中可逆残差网络的实现

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万