过滤器相似度
CIFAR10模型砝码
使用链接下载权重(您必须手动下载和提取)。 要将路径设置为下载的weights文件夹,请在vgg.py指定path_to_state_dict 。
依存关系
通过运行以下命令安装依赖项:
pip install -r requirements.txt
选项:
-net -架构的选择(默认:resnet18) -dataset选择数据集(默认值:cifar10) -tr-batch训练批次大小(默认值:512) -val-batch验证批次大小(默认值:512) -lr学习率(默认值:0.1) -wd权重衰减(默认值:5e-4) -epochs要训练的时期数(默认值:300) -cpu -cpu标志-reinit分集丢失使用标志-mode -在转移和默认训练之间选择
如何运行:
python train.py -lr 0.1 -gpu -dat
2021-02-26 12:05:46
46KB
Python
1