jalf/TIR_PROJ
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Approach refinement

Browse Source
This commit is contained in:
Johnny Fernandes
2026-04-26 02:19:10 +01:00
parent 287743709a
commit d1aab20322
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/train.py
+9 -5
View File
@@ -380,8 +380,8 @@ DEFAULT_CONFIG = {
"W_COMPLETE": 100.0, "W_COMPLETE": 100.0,
"W_STEP_COST": 0.02, "W_STEP_COST": 0.02,
"W_COMPACT": 0.0, "W_COMPACT": 0.0,
"W_WALL_TOUCH": 0.15, "W_WALL_TOUCH": 0.04,
"WALL_TOUCH_BUFFER": 0.8, "WALL_TOUCH_BUFFER": 0.3,
"ALIGN_SHAPE": "standoff", "ALIGN_SHAPE": "standoff",
"ALIGN_GATED": True, "ALIGN_GATED": True,
"ENTRY_AWARE": False, "ENTRY_AWARE": False,
@@ -408,11 +408,15 @@ def parse_args():
def main(): def main():
args = parse_args() args = parse_args()
# Load config # Load config: --config overrides, else auto-load config.json if present
cfg = dict(DEFAULT_CONFIG) cfg = dict(DEFAULT_CONFIG)
if args.config: config_path = args.config
with open(args.config) as f: if config_path is None and os.path.exists("config.json"):
config_path = "config.json"
if config_path:
with open(config_path) as f:
cfg.update(json.load(f)) cfg.update(json.load(f))
print(f"Config loaded from {config_path}")
rcfg = {k: v for k, v in cfg.items() if hasattr(HerdingEnv, k)} rcfg = {k: v for k, v in cfg.items() if hasattr(HerdingEnv, k)}