當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python mxnet.autograd.record用法及代碼示例

用法:

mxnet.autograd.record(train_mode=True)

參數：

train_mode：(bool, default True) - 前向傳遞是處於訓練模式還是預測模式。這控製了一些層的行為，例如 Dropout，BatchNorm

返回要在 ‘with’ 語句中使用的 autograd 記錄範圍上下文，並捕獲需要計算梯度的代碼。

注意：

使用train_mode=False進行轉發時，對應的反向也應使用train_mode=False，否則梯度未定義。

例子：

with autograd.record():
    y = model(x)
    backward([y])
metric.update(...)
optim.step(...)

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.autograd.record。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。