Bering.models.GaussianSmearing.requires_grad_

GaussianSmearing.requires_grad_(requires_grad=True)

Change if autograd should record operations on parameters in this module.

This method sets the parameters’ requires_grad attributes in-place.

This method is helpful for freezing part of the module for finetuning or training parts of a model individually (e.g., GAN training).

See locally-disable-grad-doc for a comparison between .requires_grad_() and several similar mechanisms that may be confused with it.

Return type:

TypeVar(T, bound= Module)

Args:

requires_grad (bool): whether autograd should record operations on

parameters in this module. Default: True.

Returns:

Module: self