If I am running only evaluation and not training, there are errors as tr_loss and nb_tr_steps are undefined.