文件名称:pytorch-trainer:PyTorch的培训师像Chainer的培训师
文件大小:140KB
文件格式:ZIP
更新时间:2024-05-14 10:04:34
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