add option test_phase_period to ClassificationTask

classy_vision/tasks/classification_task.py

@@ -107,6 +107,7 @@ def __init__(self):
         self.datasets = {}
         self.meters = []
         self.num_epochs = 1
+        self.test_phase_period = 1
         self.test_only = False
         self.base_model = None
         self.optimizer = None
@@ -146,6 +147,15 @@ def set_num_epochs(self, num_epochs: Union[int, float]):
         self.num_epochs = num_epochs
         return self
 
+    def set_test_phase_period(self, test_phase_period: int):
+        """Set the period of test phase.
+
+        Args:
+            test_phase_period: The period of test phase
+        """
+        self.test_phase_period = test_phase_period
+        return self
+
     def set_dataset(self, dataset: ClassyDataset, phase_type: str):
         """Set dataset for phase type on task
 
@@ -295,6 +305,7 @@ def from_config(cls, config: Dict[str, Any]) -> "ClassificationTask":
         task = (
             cls()
             .set_num_epochs(config["num_epochs"])
+            .set_test_phase_period(config.get("test_phase_period", 1))
             .set_loss(loss)
             .set_test_only(test_only)
             .set_model(model)
@@ -378,8 +389,11 @@ def _build_phases(self):
             phases = [{"train": True} for _ in range(self.num_epochs)]
 
             final_phases = []
-            for phase in phases:
+            for i, phase in enumerate(phases):
                 final_phases.append(phase)
+                if (i + 1) % self.test_phase_period == 0:
+                    final_phases.append({"train": False})
+            if final_phases[-1]["train"]:
                 final_phases.append({"train": False})
             return final_phases