【文件属性】:
文件名称:pytorch-trainer:PyTorch的培训师像Chainer的培训师
文件大小:140KB
文件格式:ZIP
更新时间:2021-04-24 16:17:54
Python
PyTorch的培训师像Chainer的培训师
我们可以在PyTorch上使用Trainer,Evaluator,Extension和Reporter。
安装
pip install git+https://github.com/Hiroshiba/pytorch-trainer
例子
请参阅是从modifyed 。
# Train with Trainer
PYTHONPATH= ' . ' python examples/train_mnist.py \
--device cuda \
--autoload \
--epoch 5
LogReport扩展的日志:
epoch main/loss validation/main/loss main/accuracy validation/main/accuracy elapsed_time
0