Skip to content

Commit

Permalink
offline did not use cosine learning rate decay
Browse files Browse the repository at this point in the history
  • Loading branch information
ssnl committed Aug 11, 2023
1 parent f3cce80 commit 4f11323
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions offline/run_maze2d.sh
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,11 @@ args=(
agent.quasimetric_critic.losses.latent_dynamics.weight=1
# critic
agent.quasimetric_critic.losses.critic_optim.lr=5e-4
agent.quasimetric_critic.losses.critic_optim.cosine_lr_decay_final_mul=0
agent.quasimetric_critic.losses.global_push.softplus_beta=0.01
agent.quasimetric_critic.losses.global_push.softplus_offset=500
# actor
agent.actor.model.arch="[1024,1024,1024,1024]"
agent.actor.losses.actor_optim.lr=3e-5
agent.actor.losses.actor_optim.cosine_lr_decay_final_mul=0
agent.actor.losses.min_dist.adaptive_entropy_regularizer=False
agent.actor.losses.min_dist.add_goal_as_future_state=False
agent.actor.losses.behavior_cloning.weight=0.05
Expand Down

0 comments on commit 4f11323

Please sign in to comment.