fl_sim.models.reset_parameters

fl_sim.models.reset_parameters#

fl_sim.models.reset_parameters(module: Module) None[source]#

Reset the parameters of a module and its children.

Parameters:

module (torch.nn.Module) – The module to reset.