chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch:入门与实战》)
MIT License
12.01k stars 3.73k forks source link

第六章 救急,如何运行训练测试,小白请教 #187

Open PingYufeng opened 5 years ago

PingYufeng commented 5 years ago

请教如何在pycharm中运行下面的训练和测试命令?!!


训练 必须首先启动visdom:

python -m visdom.server 然后使用如下命令启动训练:

在gpu0上训练,并把可视化结果保存在visdom 的classifier env上

python main.py train --data-root=./data/train --use-gpu=True --env=classifier 详细的使用命令可使用

python main.py help

测试

python main.py --data-root=./data/test --use-gpu=False --batch-size=256


shazhongcheng commented 5 years ago

python -m visdom.server 这个要在命令行运行

python main.py train --data-root=./data/train --use-gpu=True --env=classifier 参数这样 image image image

测试同,具体参数怎么输 你可以百度,不行谷歌,介绍都很详细