轻量级PyTorch通用训练模板pytorch-accelerated解析:1 -- API之Trainer
简介
从这一节开始,详细看一下pytorch-accelerated的API。
本文是对Trainer的API的解析。
Trainer概览
训练器Trainer用来封装特定任务的整个训练循环,将模型、损失函数和优化器结合在一起,并为训练过程的每一步提供执行行为规范。
Trainer的实现是这样的:它提供了训练部分的(可重复的)实现,这些部分在被定义后很少发生变化——比如创建数据加载器,或如何将一批数据送入模型——同时与可能发生变化的部分保持解耦,比如模型、数据集、损失函数和优化器。
1
2
3
4
5
6
7
8
9
class Trainer:
def __init__(