Fixed bug tto check if log_writer is not None before calling log_model

eae14cab · Manan Goel · f1610456 · eae14cab
Commit eae14cab authored Apr 15, 2022 by Manan Goel
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

tools/program.py tools/program.py +4 -3

No files found.
--- a/tools/program.py
+++ b/tools/program.py
@@ -380,6 +380,7 @@ def train(config,
                epoch=epoch,
                global_step=global_step)

+            if log_writer is not None:
                log_writer.log_model(is_best=False, prefix="latest")

        if dist.get_rank() == 0 and epoch > 0 and epoch % save_epoch_step == 0:
@@ -394,7 +395,7 @@ def train(config,
                best_model_dict=best_model_dict,
                epoch=epoch,
                global_step=global_step)
-
+            if log_writer is not None:
                log_writer.log_model(is_best=False, prefix='iter_epoch_{}'.format(epoch))

    best_str = 'best metric, {}'.format(', '.join(