Renamed should_stop to has_stopped in roi_loader.

Summary: Renamed should_stop to has_stopped in roi_loader. Reviewed By: rbgirshick Differential Revision: D9596732 fbshipit-source-id: b34a5b20d215fa7423d96c9a8a9cf271b500b530
xzhewei · Dec 5, 2018 · 2f3d2f7 · 2f3d2f7
1 parent b44dfdd
commit 2f3d2f7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/detectron/roi_data/loader.py b/detectron/roi_data/loader.py
@@ -241,7 +241,7 @@ def start(self, prefill=False):
                     self.shutdown()
                     break
 
-    def should_stop(self):
+    def has_stopped(self):
         return self.coordinator.should_stop()
 
     def shutdown(self):

diff --git a/detectron/utils/train.py b/detectron/utils/train.py
@@ -60,7 +60,7 @@ def train_model():
     CHECKPOINT_PERIOD = int(cfg.TRAIN.SNAPSHOT_ITERS / cfg.NUM_GPUS)
 
     for cur_iter in range(start_iter, cfg.SOLVER.MAX_ITER):
-        if model.roi_data_loader.should_stop():
+        if model.roi_data_loader.has_stopped():
             handle_critical_error(model, 'roi_data_loader failed')
         training_stats.IterTic()
         lr = model.UpdateWorkspaceLr(cur_iter, lr_policy.get_lr_at_iter(cur_iter))