dogs vs cats:《深度学习框架Pytorch入门与实践》第六章原始代码 源码
PyTorch实践指南 本文是文章配套代码,请参照或者对应更好的了解而文件组织和代码细节。 数据下载 从下载所需的数据;或者直接从此下载培训和 解压并把训练集和测试集分别放在一个文件夹中 安装 PyTorch:可按照的指南,根据自己的平台安装指定的版本 安装指定依赖: pip install -r requirements.txt 训练 必须首先启动visdom: python -m visdom.server 然后使用如下命令启动训练: # 在gpu0上训练,并把可视化结果保存在visdom 的classifier env上 python main.py train --train-data-root=./data/train --use-gpu --env=classifier 详细的使用命令可使用 python main.py help 测试 python main.py te
文件列表
dogs-vs-cats-master.zip
(预估有个15文件)
dogs-vs-cats-master
models
basic_module.py
1KB
resnet34.py
2KB
__init__.py
191B
squeezenet.py
991B
alexnet.py
1KB
main.py
5KB
PyTorch实战指南.md
30KB
data
暂无评论