MnistModel.requires_grad_

MnistModel.requires_grad_(requires_grad: bool = True) → T

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 disabling gradient computation for a comparison between .requires_grad_() and several similar mechanisms that may be confused with it.

Args:

requires_grad (bool): whether autograd should record operations on

parameters in this module. Default: True.

Returns:

Module: self

previous

MnistModel.register_parameter

next

MnistModel.set_extra_state