This function is used to define hyper-parameters for the optimizer initialization method.
Arguments
- module
An
nn_module
that has beensetup()
.- ...
The parameters passed here will be used to initialize the optimizers. For example, if your optimizer is
optim_adam
and you passlr=0.1
, then theoptim_adam
function is called withoptim_adam(parameters, lr=0.1)
when fitting the model.
See also
Other set_hparam:
set_hparams()