jakegrigsby commited on
Commit
02dccd4
·
verified ·
1 Parent(s): 2894640

Upload folder using huggingface_hub

Browse files
Files changed (32) hide show
  1. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_0.pt +3 -0
  2. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_10.pt +3 -0
  3. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_12.pt +3 -0
  4. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_14.pt +3 -0
  5. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_16.pt +3 -0
  6. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_18.pt +3 -0
  7. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_2.pt +3 -0
  8. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_20.pt +3 -0
  9. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_22.pt +3 -0
  10. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_24.pt +3 -0
  11. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_26.pt +3 -0
  12. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_28.pt +3 -0
  13. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_30.pt +3 -0
  14. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_32.pt +3 -0
  15. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_34.pt +3 -0
  16. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_36.pt +3 -0
  17. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_38.pt +3 -0
  18. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_4.pt +3 -0
  19. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_40.pt +3 -0
  20. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_42.pt +3 -0
  21. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_44.pt +3 -0
  22. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_46.pt +3 -0
  23. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_48.pt +3 -0
  24. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_50.pt +3 -0
  25. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_52.pt +3 -0
  26. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_54.pt +3 -0
  27. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_56.pt +3 -0
  28. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_58.pt +3 -0
  29. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_6.pt +3 -0
  30. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_60.pt +3 -0
  31. synthetic-rl-v1/ckpts/policy_weights/policy_epoch_8.pt +3 -0
  32. synthetic-rl-v1/config.txt +106 -0
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:138e3b239c53d43b0e322af7e7250c76f3b486093d7da027baec4f1ae32f7f36
3
+ size 802136150
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_10.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5e263f3bfd9522b50522bfd0772132dd8568f81918032ed41e3f63610875837
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_12.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbd2d38295eb7fc29529cf916a656c7cbf12a34754a626a330e7ee4e56f17f76
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_14.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdcc04411bdb88da06e6b1c2a9295b7d5ccbba46eaba527a3d0c75a2b44bfe19
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_16.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dce4d292d4ce71f6715aa308476cf5af3935f14f5e4831af50e01c5d86aae10
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_18.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e54ab381bd31bfc04864ce016798a6219077310f675d264a4d0b68895b892adb
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d18edeed09295978633c8fbd58da8af833447e076e8c33895d5cd7909e34e60
3
+ size 802136150
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_20.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:071ab72dab5c4b0150eee2842152382ed0c62fe68f346bea8af6c3f8159c86b0
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_22.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d85424f297c157efb5420eb3465f61d2c15fe0e9634325729fab7fec9fde57b
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_24.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca924bd0d38b43deb9267a024d26834aaaf48805289f311b865143da0af5bfaa
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_26.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90aa69350f2f4c9c3ce92115b23369294991c6a3c35e6b4ff986c801b8dd7032
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_28.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3ece160d904969e1daaea639131c2c3b1d3ad735509710fd12357508fdacdf0
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_30.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6313ec949f048bece17a0ff24af17dfccc972f8eb360809410cdb0bc65059a7
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_32.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:805e68fdaa826e76de2bf4fb80ad8d830895c72a4db58298e7e93affd19d9f8f
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_34.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c394c6f0c483dbe3e1a8124ba8829d22644c9e552781e14d87078531e83a91c
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_36.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:822f024359cac5457a08e2f5c05cf59b92c05eb205bc5fab4007a28020567275
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_38.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c4bffddb1d37b2cde0a6a67866d0f05a7389e06962a522c00d1dbc09f645a1d
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e06d8b4210a9260d23c352a72a1165bda9ec6c126b79c4a164fb7acc1ed8d8a
3
+ size 802136150
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_40.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:282038907cad0decd2e83f6c4442a64ba6e183bffd7da2cb4befb307704e2b82
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_42.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5fe3d3629ff7b659ee4b6a136185ae01d2490800348e956bf5d37a56b1470bd
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_44.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fcbff6a9b040d0a43de88dd86415e8e678ce0359c00261cfc4205ce8e4f73ac
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_46.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9580944f43d2b3c8e0d9304ca801254afcc26da70888b24df3d78529758122c8
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_48.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14f1f0b21a687d8e6ab63dd3895a3f9fd3d32f8bd05b84dd556b2b55c3d70d89
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_50.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4317c9a98cc7b7b455d0b536f242e77947642ab5dae82656945f98228af4df8f
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_52.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a2ea534fa0dcbb5adaac6ac3425f6d9ddb836e91dc5adf0d6ca8e9e8afd78f5
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_54.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68714c587dfc9aca4959c117f7971ced78b8d33b63b4b6e1435f5d7770f80a5c
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_56.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad4ecb377c9c29e8ea25b73eea96866f24e85a80a326a053dd007f405a31f905
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_58.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c117578c1778be1c6d722c757b20713b45afcc882d949ee5d0a62cc48853526b
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59f67d0f27dbc53a31d7d8b7018b16637f835351e88f3158e840096a15b02037
3
+ size 802136150
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_60.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b93007ea01c3ed233545f328fedcf726d1b6372c119aa009bab53e461d201f82
3
+ size 802136514
synthetic-rl-v1/ckpts/policy_weights/policy_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d18e7ac8d7fe823525d3d15ccbfa8b22f3f482a32c89c9e1b56109f506f31ef
3
+ size 802136150
synthetic-rl-v1/config.txt ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Parameters for Actor:
2
+ # ==============================================================================
3
+ Actor.activation = 'leaky_relu'
4
+ Actor.cont_dist_kind = 'normal'
5
+ Actor.d_hidden = 512
6
+ Actor.dropout_p = 0.0
7
+ Actor.gmm_modes = 5
8
+ Actor.log_std_high = 2.0
9
+ Actor.log_std_low = -5.0
10
+ Actor.n_layers = 2
11
+
12
+ # Parameters for Agent:
13
+ # ==============================================================================
14
+ Agent.fake_filter = False
15
+ Agent.gamma = 0.999
16
+ Agent.num_critics = 6
17
+ Agent.num_critics_td = 2
18
+ Agent.offline_coeff = 1.0
19
+ Agent.online_coeff = 0.0
20
+ Agent.popart = True
21
+ Agent.reward_multiplier = 10.0
22
+ Agent.tau = 0.003
23
+ Agent.use_multigamma = True
24
+ Agent.use_target_actor = True
25
+
26
+ # Parameters for Experiment:
27
+ # ==============================================================================
28
+ Experiment.batches_per_update = 1
29
+ Experiment.critic_loss_weight = 10.0
30
+ Experiment.env_mode = 'async'
31
+ Experiment.force_reset_train_envs_every = None
32
+ Experiment.grad_clip = 1.0
33
+ Experiment.has_replay_buffer_rights = True
34
+ Experiment.l2_coeff = 0.001
35
+ Experiment.learning_rate = 0.0001
36
+ Experiment.local_time_optimizer = False
37
+ Experiment.lr_warmup_steps = 500
38
+ Experiment.mixed_precision = 'no'
39
+ Experiment.padded_sampling = 'none'
40
+ Experiment.save_trajs_as = 'npz'
41
+ Experiment.stagger_traj_file_lengths = True
42
+ Experiment.wandb_group_name = None
43
+
44
+ # Parameters for FlashAttention:
45
+ # ==============================================================================
46
+ FlashAttention.window_size = (-1, -1)
47
+
48
+ # Parameters for MetamonTstepEncoder:
49
+ # ==============================================================================
50
+ MetamonTstepEncoder.d_model = 160
51
+ MetamonTstepEncoder.extra_emb_dim = 18
52
+ MetamonTstepEncoder.n_heads = 8
53
+ MetamonTstepEncoder.n_layers = 5
54
+ MetamonTstepEncoder.scratch_tokens = 11
55
+ MetamonTstepEncoder.token_mask_aug = False
56
+
57
+ # Parameters for Multigammas:
58
+ # ==============================================================================
59
+ Multigammas.continuous = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
60
+ Multigammas.discrete = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
61
+
62
+ # Parameters for MultiModalEmbedding:
63
+ # ==============================================================================
64
+ MultiModalEmbedding.dropout = 0.05
65
+ MultiModalEmbedding.numerical_tokens = 6
66
+
67
+ # Parameters for NCritics:
68
+ # ==============================================================================
69
+ NCritics.activation = 'leaky_relu'
70
+ NCritics.d_hidden = 512
71
+ NCritics.dropout_p = 0.0
72
+ NCritics.n_layers = 2
73
+
74
+ # Parameters for PopArtLayer:
75
+ # ==============================================================================
76
+ PopArtLayer.beta = 0.0005
77
+ PopArtLayer.init_nu = 100.0
78
+
79
+ # Parameters for TformerTrajEncoder:
80
+ # ==============================================================================
81
+ TformerTrajEncoder.activation = 'leaky_relu'
82
+ TformerTrajEncoder.causal = True
83
+ TformerTrajEncoder.d_ff = 5120
84
+ TformerTrajEncoder.d_model = 1280
85
+ TformerTrajEncoder.dropout_attn = 0.0
86
+ TformerTrajEncoder.dropout_emb = 0.05
87
+ TformerTrajEncoder.dropout_ff = 0.05
88
+ TformerTrajEncoder.dropout_qkv = 0.0
89
+ TformerTrajEncoder.head_scaling = True
90
+ TformerTrajEncoder.n_heads = 20
91
+ TformerTrajEncoder.n_layers = 9
92
+ TformerTrajEncoder.norm = 'layer'
93
+ TformerTrajEncoder.normformer_norms = True
94
+ TformerTrajEncoder.sigma_reparam = True
95
+
96
+ # Parameters for TimestepTransformer:
97
+ # ==============================================================================
98
+ # None.
99
+
100
+ # Parameters for TokenEmbedding:
101
+ # ==============================================================================
102
+ # None.
103
+
104
+ # Parameters for TransformerTurnEmbedding:
105
+ # ==============================================================================
106
+ TransformerTurnEmbedding.dropout = 0.05