disable_dropout

torchtune.training.disable_dropout(model: Module) → None

禁用给定模型中的 dropout 层。

参数:

model (torch.nn.Module) – 应禁用 dropout 层的模型。