In my epoch loop I have train and validate as separate function calls. In the train function I call model.train() at the start, and at the start of the validate function I call model.eval(). I thought this would automatically handle making the model trainable again…however when I comment out the validate function (so the model only trains) it actually seems to work.
Do I need to run model.train() somewhere again?