The code should generally work.
Your model definition of RMPPE has a typo in its __init__ method, where you are using three underscores on the right hand side. If you fix this, model_pose will also contain the submodules of this model.
Also, in torch.cat you are using dims which should be dim