Solved by adrianjav in post #2 
              
              
                You could do something similar to what is done when loading the VGG19 network in this tutorial. 
Create your CNN model with two attributes, train it and then just use the part you’re interested in. Something like: 
def MyCNNNetwork(nn.Module):
    ...

def MyDenseNetwork(nn.Module):
     ...

def My…