pytorch-trainer:PyTorch的培训师像Chainer的培训师

时间:2024-05-14 10:04:34
【文件属性】:

文件名称: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


网友评论