\u200E

[ torch 参数更多 ]torch.nn.Module.train

torch.nn.Module.train 

          torch.nn.Module.train(mode=True)

         

paddle.nn.Layer.train 

          paddle.nn.Layer.train()

         

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射

PyTorch	PaddlePaddle	备注
`mode`	无	设置训练模式，Paddle 无此参数，暂无转写方式。