set by_epoch=False

mmseg/core/evaluation/eval_hooks.py

@@ -8,6 +8,9 @@ class EvalHook(BasicEvalHook):
     """Single GPU EvalHook, with efficient test support.
 
     Args:
+        by_epoch (bool): Determine perform evaluation by epoch or by iteration.
+            If set to True, it will perform by epoch. Otherwise, by iteration.
+            Default: False.
         efficient_test (bool): Whether save the results as local numpy files to
             save CPU memory during evaluation. Default: False.
     Returns:
@@ -16,8 +19,8 @@ class EvalHook(BasicEvalHook):
 
     greater_keys = ['mIoU', 'mAcc', 'aAcc']
 
-    def __init__(self, *args, efficient_test=False, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self, *args, by_epoch=False, efficient_test=False, **kwargs):
+        super().__init__(*args, by_epoch=by_epoch, **kwargs)
         self.efficient_test = efficient_test
 
     def after_train_iter(self, runner):
@@ -43,6 +46,9 @@ class DistEvalHook(BasicDistEvalHook):
     """Distributed EvalHook, with efficient test support.
 
     Args:
+        by_epoch (bool): Determine perform evaluation by epoch or by iteration.
+            If set to True, it will perform by epoch. Otherwise, by iteration.
+            Default: False.
         efficient_test (bool): Whether save the results as local numpy files to
             save CPU memory during evaluation. Default: False.
     Returns:
@@ -51,8 +57,8 @@ class DistEvalHook(BasicDistEvalHook):
 
     greater_keys = ['mIoU', 'mAcc', 'aAcc']
 
-    def __init__(self, *args, efficient_test=False, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self, *args, by_epoch=False, efficient_test=False, **kwargs):
+        super().__init__(*args, by_epoch=by_epoch, **kwargs)
         self.efficient_test = efficient_test
 
     def after_train_iter(self, runner):