文件名称:HyperNet:适用于ResNet的HyperNetworks的PyTorch实施(Ha等人,ICLR 2017)
文件大小:6KB
文件格式:ZIP
更新时间:2024-04-08 17:27:03
machine-learning deep-learning hypernetworks Python
超网络 适用于ResNet的PyTorch实施(Ha等人,ICLR 2017)。该代码主要用于CIFAR-10和CIFAR-100,但是将其用于任何其他数据集都非常容易。将其用于不同深度的ResNet架构也非常容易。 我们使用pytorch闪电来控制整个管道。 怎么跑 python train.py --dataset {cifar10/cifar100} --gpus $num_gpu -j $num_workers --distributed_backend ddp 已过期。此回购协议已将HyperNet修改为逐层实施,使用起来更加方便。需要注意的是,我们发现尽管作者设置了in_size和out_size 。实际上,您应该将in_size和out_size设置为16,否则将失败。
【文件预览】:
HyperNet-main
----.gitignore(1KB)
----requirements.txt(85B)
----README.md(938B)
----resnet.py(7KB)
----train.py(9KB)