Let training run until max_epochs when fine-tuning

Setting --train.resume True will only resume for a single epoch (because of this line).

We should probably delete this line and let the training run until train.max_epochs is reached.