are your custom parameters registered in the constructor with self.param_name = nn.Parameter(...)?
The key is to wrap them in nn.Parameter, or else they wont be under the purview of state_dict or model.parameters()