Sorry, model_fit is only a custom function to do the following: loss = F.nll_loss(x_pred, x_output, ignore_index). So, what I understand is that I do not need to call scheduler.step() when performing the evaluation process.