12、构建 StepByStep 类进行模型训练

构建 StepByStep 类进行模型训练

1. 从空类开始

最初，我们有一个完全空的类 StepByStep ：

class StepByStep(object):
    pass

这个类目前没什么用，接下来我们要让它变得更有趣。

2. 构造函数

构造函数 __init__ 是构建类的起点，它定义了类的组成部分，也就是类的属性。典型的属性包括：
- 用户提供的参数
- 创建时不可用的其他对象的占位符（类似于延迟参数）
- 我们可能想要跟踪的变量
- 使用一些参数和高阶函数动态构建的函数

2.1 参数

用户需要指定的主要参数有三个：优化器、损失函数和模型。另外，还需要一个用于训练模型的设备信息。我们会自动检查是否有可用的 GPU，如果没有则使用 CPU，但也提供了一个 to() 方法让用户可以指定不同的设备。

构造函数 __init__() 最初的样子如下：

class StepByStep(object):
    def __init__(self, model, loss_fn, optimizer):
        # 这里定义类的属性
        # 首先将参数存储为属性以便后续使用
        self.model = model