How to change the class name when loading a pre-trained model?

1

Hi, Ive trained two different models seperately in a sperate file, which both include the same class name but different architecture for that class. For example:

class CNN(nn.Module):
     some stuff

class CNN(nn.Module):
     some other stuff (other than the first)

I need to load and run the 2 models together. How can I do this? Which class will be assigned to which and how can i change the class name in that case?

2

This is related to Python itself, and not PyTorch, but have a look here!