Thank you @shivammehta007
But how to call these two models?  I am using this function for training
Losses, Accuracies = fp.fit_sm(model=model, optimizer=optimizer, epochs=100,                                    trainloader=tr_loader, validloader=ts_loader,
                                   criterion=nn.CrossEntropyLoss(),
                                   device=device, verbose=True)


Here fp.fit_sm is a function for model training,
I got these three models , model 1 , model 2 and merged model, but which model should i call here in this above snippet,I  am blank here ,how to call these function here plz guide.