Fix the racing issue in RL strategy submit model (#5135)

e954774e · Yuge Zhang · GitHub · aa98462d · e954774e
Unverified Commit e954774e authored Oct 12, 2022 by Yuge Zhang Committed by GitHub Oct 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

nni/nas/strategy/_rl_impl.py nni/nas/strategy/_rl_impl.py +2 -2

No files found.
--- a/nni/nas/strategy/_rl_impl.py
+++ b/nni/nas/strategy/_rl_impl.py
@@ -136,8 +136,8 @@ class ModelEvaluationEnv(gym.Env[ObservationType, int]):
        if self.cur_step == self.num_steps:
            with _thread_lock:
                model = get_targeted_model(self.base_model, self.mutators, self.sample)
-            _logger.info(f'New model created: {self.sample}')
+                _logger.info(f'New model created: {self.sample}')
-            submit_models(model)
+                submit_models(model)
            wait_models(model)
            if model.status == ModelStatus.Failed:
                return self.reset(), 0., False, {}