Solved by ptrblck in post #2 
              
              
                Yes, you can write own scripts for the training and evaluation of the model. Make sure to load the state_dict properly before starting the evaluation. 


Yes, that should be possible.