model.train()

`model.train()` 是 PyTorch 中的一个方法，用于将模型设置为训练模式。在训练模式下，模型会保持跟踪梯度并进行反向传播。相反，在评估模式下，模型不会跟踪梯度，通常用于推断或测试阶段。如果你的模型包含了 Dropout 或 BatchNorm 层，那么你需要在训练和测试阶段分别调用 `model.train()` 和 `model.eval()` 方法，以确保网络行为的一致性。