【文件属性】:
文件名称:BossNAS:老板NAS
文件大小:170KB
文件格式:ZIP
更新时间:2021-03-25 13:04:56
Python
老板NAS
该存储库包含PyTorch评估代码,再培训代码和我们的论文预训练模型: BossNAS:通过逐块智能自我监督神经体系结构搜索探索混合CNN变压器。 [ ]
训练与合奏引导的暹罗超网的插图。
具有灵活下采样位置的类似织物的混合CNN变压器搜索空间的图示。
我们的结果和训练有素的模型
以下是我们搜索的模型的摘要:
模型
MAdds
步进时间
前1名(%)
前5 (%)
网址
没有SE的BossNet-T0
3.4B
101毫秒
80.5
95.0
BossNet-T0
3.4B
115毫秒
80.8
95.2
BossNet-T0 ^
5.7倍
147毫秒
81.6
95.6
同上
BossNet-T1
7.9B
156毫秒
81.9
95.6
BossNet-T1 ^
10.5B
165毫秒
82.2
95.7
同上
这是我们方法的体
【文件预览】:
BossNAS-main
----NATS_SS_ranking()
--------get_model_score_nats.py(2KB)
--------path_rank()
--------nats_cifar10_acc_rank.yml(718KB)
----HyTra_retraining()
--------main.py(16KB)
--------engine.py(4KB)
--------utils.py(7KB)
--------boss_models.py(8KB)
--------datasets.py(4KB)
--------samplers.py(2KB)
--------boss_candidates()
----MBConv_ranking()
--------path_rank()
--------get_model_score_mbconv.py(4KB)
----README.md(6KB)