aa51e79cdbcbedbedeef68bcef646b2d43993753,src/sdk/pynni/nni/ppo_tuner/ppo_tuner.py,PPOTuner,_actions_to_config,#PPOTuner#Any#,492

Before Change

                else:
                    chosen_arch[block_name][layer_name][key] = [act_name]
            elif key == "chosen_layer":
                chosen_arch[block_name][layer_name][key] = act_name
            else:
                raise ValueError("unrecognized key: {0}".format(key))
        return chosen_arch

After Change

            (_key, _type) = self.actions_to_config[cnt]
            if _type == "input_choice":
                if act_name == "None":
                    chosen_arch[_key] = {"_value": [], "_idx": []}
                else:
                    candidates = self.search_space[_key]["_value"]["candidates"]
                    idx = candidates.index(act_name)
                    chosen_arch[_key] = {"_value": [act_name], "_idx": [idx]}
            elif _type == "layer_choice":
                idx = self.search_space[_key]["_value"].index(act_name)
                chosen_arch[_key] = {"_value": act_name, "_idx": idx}
            else:
                raise ValueError("unrecognized key: {0}".format(_type))
        return chosen_arch

Instances