超网络
适用于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,否则将失败。
1