Solved by gphilip in post #2 
              
              
                Why does it say 

? 
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().