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

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

用法:

mxnet.autograd.pause(train_mode=False)

參數：

train_mode：(bool, default False) - 是否進行訓練或預測。

返回要在‘with’ 語句中用於不需要計算梯度的代碼的範圍上下文。

例子：

with autograd.record():
    y = model(x)
    backward([y])
    with autograd.pause():
        # testing, IO, gradient updates...

相關用法


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