单路径一枪NAS
此存储库提供了Zichao Guo等人基于Pytorch的SPOS()的实现。 al。
此仓库仅包含“块搜索”以供参考。 在ImageNet上训练该网络非常耗时,这使我无法在现有资源下完成实验。 因此,此回购协议主要集中在CIFAR-10上,非常感谢Zichao Guo在某些细节上的建议。
但是,与仍有一些差异,例如数据预处理和一些超级参数。
我已经对CIFAR-10数据集进行了超网训练,并随机采样了1K模型进行验证。 模型检查点和精度分布如下:
超级网
随机搜索
环境环境
Python == 3.6.8, Pytorch == 1.1.0, CUDA == 9.0.176, cuDNN == 7.3.0, GPU == Single GTX 1080Ti
数据集
SPOS可以直接在CIFAR-10和ImageNet上进行训练。 可以使用此代码自动下载CIFA
2021-09-22 18:33:07
101KB
Python
1