| from __future__ import annotations |
|
|
| DEFAULT_TRAIN_RUN_NAME = "curriculum_consolidate_vertical_v2_stage5_no_leak" |
| ACTIVE_CHECKPOINT = "checkpoints/setattention_stage6plus_latest_no_leak.pt" |
|
|
| DEFAULT_CURRICULUM = "8:6:300:0.88,10:8:420:0.86,12:10:520:0.84,14:12:620:0.82" |
|
|
| TRAIN_DEFAULTS: dict[str, float | int | str] = { |
| "envs": 196608, |
| "eval_envs": 8192, |
| "steps": 64, |
| "epochs": 1, |
| "minibatches": 48, |
| "hidden": 256, |
| "depth": 3, |
| "policy_arch": "attention", |
| "planner_mode": "direct", |
| "sensor_mode": "topk", |
| "sensor_topk": 16, |
| "sensor_token_range": 9.5, |
| "sensor_fov_degrees": 120.0, |
| "sensor_token_sort": "distance", |
| "observe_progress": 0, |
| "lr": 4.0e-5, |
| "entropy_coef": 0.008, |
| "sil_coef": 0.030, |
| "capture": 8, |
| "log_every": 20, |
| "min_stage_updates": 20, |
| } |
|
|
| PHYSICS_DEFAULTS: dict[str, float] = { |
| "gravity": 30.0, |
| "accel": 65.0, |
| "ground_damping": 0.55, |
| "air_damping": 0.91, |
| "air_control": 0.10, |
| "max_speed": 4.3, |
| "jump_velocity": 7.2, |
| "sprint_speed_mult": 1.35, |
| "sprint_accel_mult": 1.25, |
| "turn_speed": 3.2, |
| "landing_pad_scale": 0.25, |
| "landing_top_tolerance": 0.03, |
| "landing_prev_slack": 0.12, |
| } |
|
|
| ENV_DEFAULTS: dict[str, float | int] = { |
| "map_height_offset": 2.85, |
| "map_vertical_scale": 1.70, |
| "map_max_height": 7.0, |
| "goal_progress_reward_scale": 1.35, |
| "first_visit_reward": 0.12, |
| "step_cost": 0.0005, |
| "fall_penalty": 4.0, |
| "success_time_bonus": 0.06, |
| } |
|
|
| MAPGEN_KWARGS: dict[str, float | int] = { |
| "map_scenic_turn_scale": 0.62, |
| "map_edge_gap_min": 0.95, |
| "map_edge_gap_max": 1.58, |
| "map_route_gap_scale": 1.22, |
| "map_route_edge_gap_floor": 0.38, |
| "map_goal_xy_min": 28.0, |
| "map_goal_xy_per_jump_min": 1.90, |
| "map_extra_span_min_frac": 0.04, |
| "map_extra_span_max_frac": 0.88, |
| "map_extra_span_end_margin": 1.25, |
| "map_platform_min_scale": 0.52, |
| "map_platform_max_scale": 1.42, |
| "map_platform_tiny_fraction": 0.18, |
| "map_platform_large_fraction": 0.18, |
| "map_platform_aspect_jitter": 0.42, |
| "map_route_tiny_fraction": 0.10, |
| "map_redirect_period": 0, |
| "map_route_detour_period": 4, |
| "map_route_detour_depth": 0.58, |
| "map_backtrack_period": 5, |
| "map_backtrack_depth": 0.42, |
| "map_hairpin_period": 7, |
| "map_hairpin_angle": 1.24, |
| "map_hairpin_depth": 0.48, |
| "map_unintuitive_depth": 0.56, |
| "map_valley_period": 5, |
| "map_valley_depth": 0.58, |
| "map_direct_decoy_fraction": 0.08, |
| "map_cluster_extra_fraction": 0.10, |
| "map_decoy_corridor_fraction": 0.06, |
| "map_false_summit_fraction": 0.14, |
| "map_false_branch_fraction": 0.20, |
| "map_false_finish_fraction": 0.16, |
| "map_greedy_trap_fraction": 0.22, |
| "map_braid_bridge_fraction": 0.16, |
| "map_braid_bridge_span": 4, |
| "map_trap_jump_fraction": 0.40, |
| "map_trap_jump_min_drop": 0.24, |
| "map_trap_jump_drop": 0.70, |
| "map_trap_jump_edge_margin": 0.04, |
| "map_trap_jump_high_fraction": 0.20, |
| "map_trap_jump_high_min_up": 0.24, |
| "map_trap_jump_high_max_up": 0.94, |
| "map_trap_jump_disguise": 0.82, |
| } |
|
|
