aryopg:

Can I just call the B model class again in A model class?


Yep.



 aryopg:

Will the B part of A weight trained when I train B separately (not trained in the same graph as A)?


Yep.
If I have correctly understood what you are proposing to do, then yes, PyTorch will handle all the complexities with its usual grace.