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 May 29, 2024, 9:54 a.m.