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 June 7, 2023, 6:19 p.m.