mlx.nn.Module.train

Module.train(mode: bool = True) → Module

将模型设置为训练模式或评估模式。

训练模式仅适用于某些层。例如，Dropout 在训练模式下应用随机掩码，但在评估模式下是恒等函数。

参数:

mode (bool) – 指示模型应处于训练模式还是评估模式。默认为：True。

返回:

更新训练模式后的模块实例。