Pre-training problem in pytorch

[Nicolo_Savioli (Nicolò Savioli)]

hey thanks have a look of re-post After transfer learning model restart from zero