giagrad.nn.Module.train

Module.train()

Sets all submodules in training mode including self.

See the documentation of the particularly modules you are using to check whether they will be affected by training/evaluation mode, e.g. Dropout, DropoutNd, etc.