Solved by ptrblck in post #6 
              
              
                You could either import the model from another script via: 
from other_file import ResNet9

model = ResNet9()

or define the model in the same script: 
def ResNet9(nn.Module):
    def __init__(self):
        ...

model = ResNet9()