如何连接 pytorch 中的模型？

Question

I made 2 classes:

class Preprocess():
    def __init
     .....
    def forward 
     ....

class Model ():
    def __init
     .....
    def forward 

I would like to do Preprocess before putting them into Model. Could you kindly tell me how can I achieve that?

Answer 1

Just compose them:

pre = Preprocess()
model = Model()

output = model(pre(input))

Answer 2

One could also use torch.nn.Sequential which has the following upsides:

• easier on the eyes
• one has to just pass the inputs instead of remembering to call Preprocess before it

Code being:

model = torch.nn.Sequential(Preprocess(), Model())