Solved by ptrblck in post #4 
              
              
                The code will output the same result as the reptrained resnet50 during evaluation (and using the same self.fc at the end for the sake of debugging). 
Code snippet: 
class PABN(nn.Module):
    def __init__(self):
        super(PABN, self).__init__()
        num_classes = 576  
      
        self.bac…