You would have to assign the module to self in the forward.
Also, make sure to register this module only once. Otherwise you would be overwriting the module and no training will be performed.
Since you are now defining new models after the __init__, you should also take care of passing these new parameters to the optimizer after all modules were created in the forward method.