Could you post your model definition please?
If you have written a custom model, you can make sure that all parameters and submodules are properly registered, by checking the model.parameters() and see if all parameters are returned.