oneflow.nn.Graph.set_grad_scaler¶ Graph.set_grad_scaler(grad_scaler: Optional[oneflow.amp.grad_scaler.GradScaler] = None)¶ Set the GradScaler for gradient and loss scaling.