Solved by AlphaBetaGamma96 in post #17 
              
              
                Not a problem @hs99! I’d suggest reading the tutorial first Saving and Loading Models — PyTorch Tutorials 1.8.1+cu102 documentation and if there are still problems, raise a new topic!