I train the pytorch model(name = A) with pre-trained model(name = B).
I want to use two models in series like this: The input is used so that the output comes out through