jakegrigsby commited on
Commit
71c8caa
·
verified ·
1 Parent(s): 70d0f0d

Upload folder using huggingface_hub

Browse files
Files changed (29) hide show
  1. small-rl-maxq/ckpts/policy_weights/policy_epoch_0.pt +3 -0
  2. small-rl-maxq/ckpts/policy_weights/policy_epoch_10.pt +3 -0
  3. small-rl-maxq/ckpts/policy_weights/policy_epoch_12.pt +3 -0
  4. small-rl-maxq/ckpts/policy_weights/policy_epoch_14.pt +3 -0
  5. small-rl-maxq/ckpts/policy_weights/policy_epoch_16.pt +3 -0
  6. small-rl-maxq/ckpts/policy_weights/policy_epoch_18.pt +3 -0
  7. small-rl-maxq/ckpts/policy_weights/policy_epoch_2.pt +3 -0
  8. small-rl-maxq/ckpts/policy_weights/policy_epoch_20.pt +3 -0
  9. small-rl-maxq/ckpts/policy_weights/policy_epoch_22.pt +3 -0
  10. small-rl-maxq/ckpts/policy_weights/policy_epoch_24.pt +3 -0
  11. small-rl-maxq/ckpts/policy_weights/policy_epoch_26.pt +3 -0
  12. small-rl-maxq/ckpts/policy_weights/policy_epoch_28.pt +3 -0
  13. small-rl-maxq/ckpts/policy_weights/policy_epoch_30.pt +3 -0
  14. small-rl-maxq/ckpts/policy_weights/policy_epoch_32.pt +3 -0
  15. small-rl-maxq/ckpts/policy_weights/policy_epoch_34.pt +3 -0
  16. small-rl-maxq/ckpts/policy_weights/policy_epoch_36.pt +3 -0
  17. small-rl-maxq/ckpts/policy_weights/policy_epoch_38.pt +3 -0
  18. small-rl-maxq/ckpts/policy_weights/policy_epoch_4.pt +3 -0
  19. small-rl-maxq/ckpts/policy_weights/policy_epoch_40.pt +3 -0
  20. small-rl-maxq/ckpts/policy_weights/policy_epoch_42.pt +3 -0
  21. small-rl-maxq/ckpts/policy_weights/policy_epoch_44.pt +3 -0
  22. small-rl-maxq/ckpts/policy_weights/policy_epoch_46.pt +3 -0
  23. small-rl-maxq/ckpts/policy_weights/policy_epoch_48.pt +3 -0
  24. small-rl-maxq/ckpts/policy_weights/policy_epoch_50.pt +3 -0
  25. small-rl-maxq/ckpts/policy_weights/policy_epoch_52.pt +3 -0
  26. small-rl-maxq/ckpts/policy_weights/policy_epoch_54.pt +3 -0
  27. small-rl-maxq/ckpts/policy_weights/policy_epoch_6.pt +3 -0
  28. small-rl-maxq/ckpts/policy_weights/policy_epoch_8.pt +3 -0
  29. small-rl-maxq/config.txt +106 -0
small-rl-maxq/ckpts/policy_weights/policy_epoch_0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:239def3b32a1244ae00b5978d43043ba4f8d65a107d8f4e7e0d52e67c145a03a
3
+ size 55821590
small-rl-maxq/ckpts/policy_weights/policy_epoch_10.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b2bc80e4b6f2bffefc4be2299e7b3234841b45f54ecb2edce439d85afa2cba7
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_12.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4222369b416d164816491595d159a7943468637fad684eab6031a57efec27e54
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_14.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6990cd796fe57ae104353e64ba88b5edb7f9fbc55c553c058d8b033f06da2ece
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_16.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:882bbcf1de6e60e14cf0175940779f8e9af40afa5eb8c33ffb28516eb916184e
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_18.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2bc15729389cbeedbe12f5e8c06a74f5eed667b5654fd3b764dae9facc7e8ec
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1ceb34b42b3dd4443ffe47aa672680ab30983a4de4e3013fe144f7abea890cd
3
+ size 55821590
small-rl-maxq/ckpts/policy_weights/policy_epoch_20.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4afd037f2e35f0bea55ee3796a49749ebe83fd96193a108c618df497346e2a06
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_22.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2e708ca1d747e87ae51719fae326c9b034ee078d174b4260dcbacaf52a0fc21
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_24.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bafaff29549e42885d4eaa14a0c6419b7eff51a44454439846fd1370e1c9e7d9
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_26.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11cfcee49602101f2bfc8768da7f1c8164afece74b2bd3a60df1db1274c73574
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_28.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5812ce2f04a83e0d9945446ca4f05c2ad56e3e726ccd53b09c0648c0018030e
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_30.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35677cd53793020e9a1988c2290473aeadadea5b3acb8eed22ad14cdb211e0bb
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_32.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32d7907f4eb47e02ff944ff04a89e3940ce1b44c3a7a6ef1d64973b8a17f4fd0
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_34.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d07dd978f2e8da2ed812e91870b0c68982f22205465d0da945349039d87e9b3a
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_36.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec10ea88e6ca3114a2604570edc23f57f663f9b128c443a38ff8b6acd899424d
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_38.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d289cd8b968ee4fdd864dcecd8443f5bd95d6ca3e3934f38c69472308dae358
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf83fc62dc6ff8865d220f4fa2ee0a2d9f5eb0fdaa91ff448f95195ed32a385d
3
+ size 55821590
small-rl-maxq/ckpts/policy_weights/policy_epoch_40.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82d02dcd00da93268278bab2f7bbfccdd1e681017194618cd937dd4c1fafe071
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_42.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2479f5f121053c3f7ca195782e2b8405cd6902db7dc0bf5b80b6375d806ccc7
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_44.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e79375b39f1e557ad12fc5dc5921608a886a37e554552d07e7678119025fb397
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_46.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a70e7762ec263b7cb19056001dcf03976b54c59e1b8a38a1dce4b4078c7b6c2
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_48.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c34227f55b81360d1a91f4dee11db86416c7bfbb4fbb0edfd5f46b5424ab4d7
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_50.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb0af867e34b189ebe1df07154381aef1517ff434547b9eb67601654648be2c9
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_52.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77f2b620d2f58c4bf777e9f6fc57277360eb120a49e0a8f6b8f95b00cbfe7260
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_54.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77d5766314a38884770184c0873ce2b7c8048bcc76d64013f29ae64d1e0e840c
3
+ size 55821762
small-rl-maxq/ckpts/policy_weights/policy_epoch_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb0035b5caa90541a125c4cafa141b061b1eb032ae60f36a8daee650aaca4982
3
+ size 55821590
small-rl-maxq/ckpts/policy_weights/policy_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6aa75f7569835955ff0ac37d982e100a7f396e9bebfc5c2cd6db4b3e6977870a
3
+ size 55821590
small-rl-maxq/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 = 300
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 = 4
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 = 100
51
+ MetamonTstepEncoder.extra_emb_dim = 18
52
+ MetamonTstepEncoder.n_heads = 5
53
+ MetamonTstepEncoder.n_layers = 3
54
+ MetamonTstepEncoder.scratch_tokens = 4
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 = 300
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 = 2048
84
+ TformerTrajEncoder.d_model = 512
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 = 8
91
+ TformerTrajEncoder.n_layers = 3
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