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 中无此参数,一般对网络训练结果影响不大。