diff --git a/angelslim/engine.py b/angelslim/engine.py index 87f813d1..5d1c45fd 100644 --- a/angelslim/engine.py +++ b/angelslim/engine.py @@ -271,6 +271,7 @@ def save( } config_dict["model_config"]["model_path"] = "Base Model Path" config_dict["global_config"]["save_path"] = "Save Model Path" + config_dict["dataset_config"]["data_path"] = "Data Path" with open(os.path.join(save_path, "angelslim_config.json"), "w") as f: json.dump(config_dict, f, indent=4)