文件名称：DS-Net:动态可压缩网络（CVPR 2021，口服）
文件大小：28KB
文件格式：ZIP
更新时间：2024-04-26 13:56:50
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)