Why does it say



 mobassir94:

Input size (MB): 0.00


?
Are you not passing anything as input to the validation step? Try printing len(val_loader) before the line print(images)  in valid_one_epoch().