Skip to content

Commit

Permalink
slight adjustments due to pre-commit, pre-commit still fails due to u…
Browse files Browse the repository at this point in the history
…nused imports, however these imports are necessary for the used environments to be registered
  • Loading branch information
MarcoMeter committed May 27, 2024
1 parent 86e59ef commit 62fea9d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cleanrl/ppo_trxl/ppo_trxl.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ class Args:
"""whether to save model into the `runs/{run_name}` folder"""

# Algorithm specific arguments
env_id: str = "MortarMayhem-Grid-v0" # MysteryPath-Grid-v0 SearingSpotlights-v0
env_id: str = "MortarMayhem-Grid-v0" # MysteryPath-Grid-v0 SearingSpotlights-v0
"""the id of the environment"""
total_timesteps: int = 200000000
"""total timesteps of the experiments"""
Expand Down Expand Up @@ -446,7 +446,7 @@ def reconstruct_observation(self):
lr = (args.init_lr - args.final_lr) * frac + args.final_lr
for param_group in optimizer.param_groups:
param_group["lr"] = lr
ent_coef = (args.init_ent_coef - args.final_ent_coef) * frac + args.final_ent_coef
ent_coef = (args.init_ent_coef - args.final_ent_coef) * frac + args.final_ent_coef

# Init episodic memory buffer using each environments' current episodic memory
stored_memories = [next_memory[e] for e in range(args.num_envs)]
Expand Down

0 comments on commit 62fea9d

Please sign in to comment.