I am training CNNs through transfer learning, following pytorch transfer learning tutorial. As shown below, after completion of training, type error occurs.

I shall be grateful, If anybody help me out.