文件名称：ist的matlab代码-pytorch-revnet:pytorch中可逆残差网络的实现
文件大小：16KB
文件格式：ZIP
更新时间：2024-06-15 01:24:16
系统开源
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万
【文件预览】：
pytorch-revnet-master
----setup.py(123B)
----.gitignore(3KB)
----param_count.py(325B)
----LICENSE(1KB)
----README.md(984B)
----tests()
--------__init__.py(0B)
--------test_revnet.py(2KB)
--------common.py(12KB)
----revnet()
--------__init__.py(984B)
--------revnet.py(19KB)
--------resnet.py(3KB)
----.gitattributes(42B)
----train_cifar.py(7KB)