autograd_set_grad_mode: Set grad mode

View source: R/autograd.R

autograd_set_grad_modeR Documentation

Set grad mode

Description

Sets or disables gradient history.

Usage

autograd_set_grad_mode(enabled)

Arguments

enabled

bool wether to enable or disable the gradient recording.


torch documentation built on Aug. 19, 2022, 5:08 p.m.