Bering.models.GCN.zero_grad

GCN.zero_grad(set_to_none=True)

Resets gradients of all model parameters. See similar function under torch.optim.Optimizer for more context.

Return type:

None

Args:

set_to_none (bool): instead of setting to zero, set the grads to None.

See torch.optim.Optimizer.zero_grad() for details.