pytorch-book icon indicating copy to clipboard operation
pytorch-book copied to clipboard

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

Open PingYufeng opened this issue 5 years ago • 1 comments

请教如何在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


PingYufeng avatar Sep 11 '19 06:09 PingYufeng

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

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

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

shazhongcheng avatar Sep 12 '19 03:09 shazhongcheng