- 下载代码之后,新建conda环境,python版本3.7,torch 和 python 对应关系
- 查看自己的cuda版本,
nvidia-smi
- 安装cuda对应的torch库,用conda安装torch库的时候把添加的源去掉,不然一直下载失败
- 弄小数据集先跑一下
train.py
【论文复现】 隐写分析 (XuNet) - Structural Design of Convolutional Neural Networks for Steganalysis- PyTorch
于 2023-07-13 14:29:03 首次发布