I copy the source code of resnet.py to the folder. And add different codes in forward(self, x) for the training case and eval case.
def forward(self, x)
      x = ......
      if self.training :
            print("...........training.......")

This print doesnt work.
from resent import resnet50
model = resnet50()
model.train() 
for i, ( imgs, labels) in enumerate(train_loader):
    prd = model(imgs)

Am I missing something?