fix a bug in filtering checkpoints (NVIDIA#6851)

Signed-off-by: Xin Yao <[email protected]>
rohitrango · Aug 8, 2023 · 1346272 · 1346272
1 parent a6e9e9f
commit 1346272
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/utils/callbacks/nemo_model_checkpoint.py b/nemo/utils/callbacks/nemo_model_checkpoint.py
@@ -96,7 +96,7 @@ def nemo_topk_check_previous_run(self):
             if checkpoint[-10:] == '-last.ckpt':
                 continue
             index = checkpoint.find(self.monitor) + len(self.monitor) + 1  # Find monitor in str + 1 for '='
-            if index != -1:
+            if index != len(self.monitor):
                 match = re.search('[A-z]', checkpoint[index:])
                 if match:
                     value = checkpoint[index : index + match.start() - 1]  # -1 due to separator hypen