0 下载源码
https://github.com/liubamboo/DSDH_PyTorch
1 下载数据集
下载cifar-10-batches-py
2 下载预训练模型
直接运行程序会自动从某个url下载模型,但是如果无法连接那个网址的话可以自行下载预训练模型然后将它放入对应的目录
/home/lwt/.cache/torch/checkpoints/
3 环境配置
安装requirements配置即可,pytorch版本为了兼容cuda版本和可以使用torch.hub,最终选择的是1.1.0版本
4 运行命令
/home/lwt/anaconda3/envs/py36-for-DSDH/bin/python /mnt/mdisk/lwt/project/DSDH_PyTorch/run.py --dataset cifar-10 --root /mnt/mdisk/lwt/datasets --gpu 0