【文件属性】：
文件名称：DS-Net:动态可压缩网络（CVPR 2021，口服）
文件大小：28KB
文件格式：ZIP
更新时间：2021-04-06 20:10:10
pruning model-compression efficient-inference dynamic-networks network-pruning
动态可压缩网络（DS-Net）
该存储库包含我们论文的PyTorch代码：（CVPR 2021 Oral）。
DS-Net的体系结构。 每个超网级的宽度通过门所预测的纤细比ρ进行自适应调整。
ImageNet上的准确性与复杂性。
用法
1.要求
安装 1.2.0+，例如：
conda install -c pytorch pytorch torchvision
安装 0.3.2，例如：
pip install timm==0.3.2
从下载ImageNet。 使用以下脚本将验证图像移动到带标签的子文件夹： :
2.第一阶段：Supernet培训
例如，使用8个GPU训练动态可瘦的MobileNet超级网（大约需要2天）：
python -m torch.distributed.launch --nproc_per_node=8 train.py /PATH/TO/Ima
【文件预览】：
DS-Net-main
----dyn_slim()
--------__init__.py(26B)
--------models()
--------apis()
--------utils()
----README.md(2KB)
----configs()
--------mobilenetv1_bn_uniform.yml(636B)
--------__init__.py(0B)
----train.py(29KB)