From 27b66ab54f9d2515211a373c7f3127d5b66b4921 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Wed, 8 Feb 2023 13:23:19 -0500 Subject: [PATCH] ability to not do val --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 9e011da..779ebb7 100644 --- a/train.py +++ b/train.py @@ -948,8 +948,7 @@ def main(args): loss_local = sum(loss_epoch) / len(loss_epoch) log_writer.add_scalar(tag="loss/epoch", scalar_value=loss_local, global_step=global_step) - # validate - validator.do_validation_if_appropriate(epoch, global_step, get_model_prediction_and_target) + validator.do_validation_if_appropriate(epoch, global_step, get_model_prediction_and_target) if validator is not None else None gc.collect() # end of epoch