From d1a83f558cff204faf5275049dd58a84c60539db Mon Sep 17 00:00:00 2001 From: Mannat Singh Date: Tue, 6 Jul 2021 20:29:00 -0700 Subject: [PATCH] Fix eval phase idx in tensorboard Differential Revision: D29577409 fbshipit-source-id: db49f88f0f43eba79bed8ddc1885ea48cbbf5d3d --- classy_vision/tasks/classification_task.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/classy_vision/tasks/classification_task.py b/classy_vision/tasks/classification_task.py index d95cb7826..eaf7200ba 100644 --- a/classy_vision/tasks/classification_task.py +++ b/classy_vision/tasks/classification_task.py @@ -1337,7 +1337,7 @@ def log_phase_end(self, tag): self.perf_log.append( { "tag": tag, - "phase_idx": self.train_phase_idx, + "phase_idx": self.train_phase_idx if self.train else self.eval_phase_idx, "im_per_sec": im_per_sec, } )