Fix 6 critical bugs in REPPO repository preventing execution

- Fix missing MUON optimizer by replacing with optax.adam
- Fix Hydra configuration parameter paths (env.name instead of env_name)
- Fix BraxGymnaxWrapper method signatures to accept params argument
- Fix training loop division by zero with proper total_time_steps
- Fix incorrect algorithm name in wandb (reppo instead of sac)
- Fix JAX key batching error in BraxGymnaxWrapper reset method
- Add comprehensive HoReKa SLURM integration with wandb logging
- Update README with detailed bug documentation and fixes
This commit is contained in:
ys1087@partner.kit.edu
2025-07-22 17:26:43 +02:00
parent 137b9e80c9
commit b240a19ceb
7 changed files with 120 additions and 17 deletions
+1 -1
View File
@@ -70,7 +70,7 @@ hyperparameters:
measure_burnin: 3
name: "sac"
name: "reppo"
seed: 0
num_seeds: 1
tune: false