vshwanilgv commited on
Commit
c4d9748
·
verified ·
1 Parent(s): 8ac1768

Upload checkpoints/train_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. checkpoints/train_config.json +19 -0
checkpoints/train_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "episode_dir": "./recorded_episodes/train",
3
+ "output_dir": "./ppo_checkpoints",
4
+ "device": "cuda:0",
5
+ "total_steps": 2000000,
6
+ "n_rollout": 2048,
7
+ "n_epochs": 4,
8
+ "batch_size": 256,
9
+ "lr": 0.0003,
10
+ "gamma": 0.99,
11
+ "gae_lambda": 0.95,
12
+ "clip_eps": 0.2,
13
+ "value_coef": 0.5,
14
+ "entropy_coef": 0.01,
15
+ "max_grad_norm": 0.5,
16
+ "save_every": 50,
17
+ "resume_from": null,
18
+ "seed": 42
19
+ }