louis_lifu
/
AReaL
mirror of https://github.com/inclusionAI/AReaL
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix var name

This commit is contained in:
kira.gw 2025-03-28 11:20:21 +08:00
parent c2b03f62d6
commit 86d08db879
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
realhf/apps/quickstart.py
View File

@ -64,8 +64,8 @@ def main():
launch_hydra_task(args["cmd"], experiment_name, trial_name, QUICKSTART_FN[args["cmd"]])
def prepare_hydra_config(name: str, PROLOGUE_path: str):
config = OmegaConf.load(PROLOGUE_path)
def prepare_hydra_config(name: str, prologue_path: str):
config = OmegaConf.load(prologue_path)
experiment_name = get_experiment_name(config.get("experiment_name"))
trial_name = get_trial_name(config.get("trial_name"))
config_dir = f"{cluster_spec.fileroot}/configs/{getpass.getuser()}/{experiment_name}/{trial_name}"

4
realhf/base/prologue.py
View File

@ -19,9 +19,9 @@ def global_init():
args = vars(parser.parse_known_args()[0])
if args[PROLOGUE_FLAG_VAR_NAME] is None:
return
PROLOGUE_path = args[PROLOGUE_FLAG_VAR_NAME]
prologue_path = args[PROLOGUE_FLAG_VAR_NAME]
config = OmegaConf.load(PROLOGUE_path)
config = OmegaConf.load(prologue_path)
external_configs = config.get(PROLOGUE_EXTERNAL_CONFIG_NAME)
if external_configs is None: