RegNet Search PyTorch:使用PyTorch搜索RegNet 源码

搜索RegNet-PyTorch 使用PyTorch和实现神经结构搜索。 在 CVPR 2020上的教程中将使用此示例和 。 模型 参考 加速器 配置 RegNet-0.4GF 官方的 72.38 RegNet-0.4GF 我们的 72.18 RegNet-4.0GF 官方的 79.03 official :使用官方配置。 ours :使用我们的搜索配置。 训练HP设置: learning rate: 0.2, batch size: 512, weight decay: 1e-4, 快速开始 安装依赖项 按照安装PyTorch。 安装其他依赖项: pip install autotorch thop torch-encoding 安装Apex(可选): git clone https://github.com/NVIDIA/apex cd apex pip install -v --n