评价此页

torch.fx.experimental.optimization.remove_dropout#

torch.fx.experimental.optimization.remove_dropout(model)[source]#

移除模块中的所有 dropout 层。

返回类型:

模块