Hi @ptrblck
You’re right in that it just hangs after it finishes executing.
I’m not too sure how to go about running this in terminal as I’ve only worked in Notebooks.  Interestingly enough, the same thing happens if I used Skorch’s NeuralNetClassifier:
image592×799 77.8 KB
Lastly, could this be related in any way to how I defined the CNN or even to how I’ve pre-processed the images?