Upload train_all_phases.py
Browse files- 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,
|
| 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:
|