Thanks for the quick reply!
I already tried that, but surprisingly, didn’t get the same output as the one I got when I first ran the training and testing loop… I thought there must be others ways to do it.
I’ll take a closer look at the code… there should be something missing.