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