fix odc hook

jeanttz · Jun 17, 2020 · 47ce648 · 47ce648
1 parent 6a5dd91
commit 47ce648
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openselfsup/hooks/odc_hook.py b/openselfsup/hooks/odc_hook.py
@@ -45,7 +45,7 @@ def after_train_iter(self, runner):
 
     def after_train_epoch(self, runner):
         # save cluster
-        if self.every_n_epochs(10) and runner.rank == 0:
+        if self.every_n_epochs(runner, 10) and runner.rank == 0:
             new_labels = runner.model.module.memory_bank.label_bank
             if new_labels.is_cuda:
                 new_labels = new_labels.cpu()