Python mxnet.autograd.train_mode用法及代码示例

用法: mxnet.autograd.train_mode()

返回要在 ‘with’ 语句中使用的范围上下文，其中前向传递行为设置为训练模式，而不更改记录状态。

例子：

y = model(x)
with autograd.train_mode():
    y = dropout(y)

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.autograd.train_mode。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。