文件名称：RegNet-Search-PyTorch:使用PyTorch搜索RegNet
文件大小：46KB
文件格式：ZIP
更新时间：2024-04-09 05:21:02
Python
搜索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
【文件预览】：
RegNet-Search-PyTorch-master
----train.py(16KB)
----configs()
--------RegNetX-4.0GF.ini(131B)
--------RegNetX-0.4GF.ini(131B)
----scripts()
--------prepare_imagenet.py(4KB)
----verify.py(6KB)
----requirements.txt(30B)
----search.py(8KB)
----LICENSE(1KB)
----gen_configs()
--------RegNet-0.4GF()
----README.md(2KB)
----test_flops.py(2KB)
----generate_configs.py(2KB)
----out_configs()
--------RegNetX-0.4GF()
----arch()
--------regnet.py(8KB)
--------base_generator.py(1KB)
--------__init__.py(0B)
----.gitignore(2KB)