nn_init_no_grad_constant_deepreg: custom nn_linear module to overwrite reset_parameters
# nn_init_constant works only if value is scalar; so warmstarts for gam does'not work
Description
custom nn_linear module to overwrite reset_parameters
# nn_init_constant works only if value is scalar; so warmstarts for gam does'not work
Usage
nn_init_no_grad_constant_deepreg(tensor, value)
Arguments
- tensor
scalar or vector
- value
value used for constant initialization