08af7771d1f1149f3da5a98bcdbf7e65abd409ba,nni/retiarii/strategies/tpe_strategy.py,TPEStrategy,run,#TPEStrategy#Any#Any#,43

Before Change

            sample_space.extend(recorded_candidates)
        self.tpe_sampler.update_sample_space(sample_space)

        try:
            _logger.info("stargety start...")
            while True:
                model = base_model
                _logger.info("apply mutators...")
                _logger.info("mutators: %s", str(applied_mutators))
                self.tpe_sampler.generate_samples(self.model_id)
                for mutator in applied_mutators:
                    _logger.info("mutate model...")
                    mutator.bind_sampler(self.tpe_sampler)
                    model = mutator.apply(model)
                // run models
                submit_models(model)
                wait_models(model)
                self.tpe_sampler.receive_result(self.model_id, model.metric)
                self.model_id += 1
                _logger.info("Strategy says: %s", model.metric)
        except Exception:
            _logger.error(logging.exception("message"))

After Change

                time.sleep(2)

            _logger.warning("num of running models: %d", len(self.running_models))
            to_be_deleted = []
            for _id, _model in self.running_models.items():
                if is_stopped_exec(_model):
                    if _model.metric is not None:
                        self.tpe_sampler.receive_result(_id, _model.metric)
                        _logger.warning("tpe receive results: %d, %s", _id, _model.metric)
                    to_be_deleted.append(_id)
            for _id in to_be_deleted:
                del self.running_models[_id]

Instances