NAS文章：Once-for-all代码分析1-代码结构

Once-for-all学习

ofa

imagenet_classification 超网训练部分

data_providers 提供数据集

elastic_nn 弹性结构

modules 弹性模块定义

dynamic_layers.py
dynamic_op.py

networks 弹性网络定义

ofa_mbv3.py
ofa_proxyless.py
ofa_resnets.py

training 渐进式训练代码

progressive_shrinking.py  #PS train核心代码

networks 正常(非弹性)网络

mobilenet_v3.py
proxyless_nets.py
resnets.py

run_manager 包装网络,提供训练时需要的一些操作

run_config.py #提供初始化数据？
run_manager.py #训练的管理
distributed_run_manager.py #分布式训练的管理

tutorial 使用once-for-all的教学，可以看看

utils 工具包，别的地方会调用

nas 搜索部分

accuracy_predictor #acc predictor
efficiency_predictor #Latency 的 predictor
search_algorithm #搜索算法核心代码
train_ofa_net.py #可以直接运行，用来fine-tune OFA超网络（progressive shrinking ）

model_zoo.py

 

参考了这篇https://blog.csdn.net/iamxiaoluoli/article/details/112394176