Solved by Franz in post #2 
              
              
                Hi, what you are doing right is to split the data into training and validation. 
You are training the model and then validating it already on unseen data (that is the idea of the validation loop). 
Your testing loop currently does exact the same, which is not needed in most cases. I think you want t…