E-Rong commited on
Commit
5acbf73
·
verified ·
1 Parent(s): 5a07374

Upload train_all_phases.py

Browse files
Files changed (1) hide show
  1. train_all_phases.py +1 -1
train_all_phases.py CHANGED
@@ -584,7 +584,7 @@ def train_phase(cfg, phase, total_timesteps, model=None):
584
  learning_rate=3e-4, n_steps=2048, batch_size=64, n_epochs=10,
585
  gamma=0.99, gae_lambda=0.95, clip_range=0.2,
586
  ent_coef=0.01, vf_coef=0.5, max_grad_norm=0.5,
587
- verbose=1, tensorboard_log="./tb_logs",
588
  device="cuda" if torch.cuda.is_available() else "cpu",
589
  )
590
  else:
 
584
  learning_rate=3e-4, n_steps=2048, batch_size=64, n_epochs=10,
585
  gamma=0.99, gae_lambda=0.95, clip_range=0.2,
586
  ent_coef=0.01, vf_coef=0.5, max_grad_norm=0.5,
587
+ verbose=1,
588
  device="cuda" if torch.cuda.is_available() else "cpu",
589
  )
590
  else: