Linear 差异对比

torch.nn.Dropout

torch.nn.Dropout(
   p=0.5,
   inplace=False
)

更多请查看 torch.nn.Dropout.

megengine.module.Dropout

megengine.module.Dropout(
   drop_prob=0.0
)

更多请查看 megengine.module.Dropout.

参数差异

drop_prob

MegEngine 中 drop_prob 参数对应 PyTorch 中的 p 参数，表示每个元素被丢弃（置为 0）的概率。

inplace

PyTorch 中 inplace 参数，表示是否要原地操作，默认不开启，MegEngine 中无此参数，一般对网络训练结果影响不大。