pytorch-book
pytorch-book copied to clipboard
第六章 救急,如何运行训练测试,小白请教
请教如何在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
python -m visdom.server 这个要在命令行运行
python main.py train --data-root=./data/train --use-gpu=True --env=classifier
参数这样
测试同,具体参数怎么输 你可以百度,不行谷歌,介绍都很详细