复现DSDH

该博客详细介绍了如何复现DSDH-pytorch项目,包括从GitHub克隆源码,下载cifar-10数据集,设置预训练模型路径,配置pytorch 1.1.0环境,以及在CUDA支持下执行python脚本的过程。然而,运行过程中遇到了未具体说明的报错信息。
摘要由CSDN通过智能技术生成

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

5 报错信息

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值