[ 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 无此参数,暂无转写方式。