Hugging Face's logo

jakegrigsby
/
metamon

Reinforcement Learning
Safetensors
amago
pokemon
game-ai
offline-rl
Model card Files
xet
Community
1
jakegrigsby commited on
Commit
fe8861f
·
1 Parent(s): 4c3e56c

add abra

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. abra/ckpts/config.txt +130 -0
  2. abra/ckpts/latest/policy.pt +3 -0
  3. abra/ckpts/policy_weights/policy_epoch_0.pt +3 -0
  4. abra/ckpts/policy_weights/policy_epoch_10.pt +3 -0
  5. abra/ckpts/policy_weights/policy_epoch_12.pt +3 -0
  6. abra/ckpts/policy_weights/policy_epoch_14.pt +3 -0
  7. abra/ckpts/policy_weights/policy_epoch_16.pt +3 -0
  8. abra/ckpts/policy_weights/policy_epoch_18.pt +3 -0
  9. abra/ckpts/policy_weights/policy_epoch_2.pt +3 -0
  10. abra/ckpts/policy_weights/policy_epoch_20.pt +3 -0
  11. abra/ckpts/policy_weights/policy_epoch_22.pt +3 -0
  12. abra/ckpts/policy_weights/policy_epoch_24.pt +3 -0
  13. abra/ckpts/policy_weights/policy_epoch_26.pt +3 -0
  14. abra/ckpts/policy_weights/policy_epoch_28.pt +3 -0
  15. abra/ckpts/policy_weights/policy_epoch_30.pt +3 -0
  16. abra/ckpts/policy_weights/policy_epoch_32.pt +3 -0
  17. abra/ckpts/policy_weights/policy_epoch_34.pt +3 -0
  18. abra/ckpts/policy_weights/policy_epoch_36.pt +3 -0
  19. abra/ckpts/policy_weights/policy_epoch_38.pt +3 -0
  20. abra/ckpts/policy_weights/policy_epoch_4.pt +3 -0
  21. abra/ckpts/policy_weights/policy_epoch_40.pt +3 -0
  22. abra/ckpts/policy_weights/policy_epoch_6.pt +3 -0
  23. abra/ckpts/policy_weights/policy_epoch_8.pt +3 -0
  24. abra/ckpts/training_states/gen9v3_epoch_0/custom_checkpoint_0.pkl +3 -0
  25. abra/ckpts/training_states/gen9v3_epoch_0/model.safetensors +3 -0
  26. abra/ckpts/training_states/gen9v3_epoch_0/optimizer.bin +3 -0
  27. abra/ckpts/training_states/gen9v3_epoch_0/random_states_0.pkl +3 -0
  28. abra/ckpts/training_states/gen9v3_epoch_0/random_states_1.pkl +3 -0
  29. abra/ckpts/training_states/gen9v3_epoch_0/random_states_2.pkl +3 -0
  30. abra/ckpts/training_states/gen9v3_epoch_0/scheduler.bin +3 -0
  31. abra/ckpts/training_states/gen9v3_epoch_10/custom_checkpoint_0.pkl +3 -0
  32. abra/ckpts/training_states/gen9v3_epoch_10/model.safetensors +3 -0
  33. abra/ckpts/training_states/gen9v3_epoch_10/optimizer.bin +3 -0
  34. abra/ckpts/training_states/gen9v3_epoch_10/random_states_0.pkl +3 -0
  35. abra/ckpts/training_states/gen9v3_epoch_10/random_states_1.pkl +3 -0
  36. abra/ckpts/training_states/gen9v3_epoch_10/random_states_2.pkl +3 -0
  37. abra/ckpts/training_states/gen9v3_epoch_10/scheduler.bin +3 -0
  38. abra/ckpts/training_states/gen9v3_epoch_12/custom_checkpoint_0.pkl +3 -0
  39. abra/ckpts/training_states/gen9v3_epoch_12/model.safetensors +3 -0
  40. abra/ckpts/training_states/gen9v3_epoch_12/optimizer.bin +3 -0
  41. abra/ckpts/training_states/gen9v3_epoch_12/random_states_0.pkl +3 -0
  42. abra/ckpts/training_states/gen9v3_epoch_12/random_states_1.pkl +3 -0
  43. abra/ckpts/training_states/gen9v3_epoch_12/random_states_2.pkl +3 -0
  44. abra/ckpts/training_states/gen9v3_epoch_12/scheduler.bin +3 -0
  45. abra/ckpts/training_states/gen9v3_epoch_14/custom_checkpoint_0.pkl +3 -0
  46. abra/ckpts/training_states/gen9v3_epoch_14/model.safetensors +3 -0
  47. abra/ckpts/training_states/gen9v3_epoch_14/optimizer.bin +3 -0
  48. abra/ckpts/training_states/gen9v3_epoch_14/random_states_0.pkl +3 -0
  49. abra/ckpts/training_states/gen9v3_epoch_14/random_states_1.pkl +3 -0
  50. abra/ckpts/training_states/gen9v3_epoch_14/random_states_2.pkl +3 -0
abra/ckpts/config.txt ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import amago.agent
2
+ import amago.experiment as amago2
3
+ import amago.nets.actor_critic as amago3
4
+ import amago.nets.traj_encoders as amago4
5
+
6
+ # Parameters for Actor:
7
+ # ==============================================================================
8
+ # None.
9
+
10
+ # Parameters for Agent:
11
+ # ==============================================================================
12
+ # None.
13
+
14
+ # Parameters for Discrete:
15
+ # ==============================================================================
16
+ Discrete.clip_prob_high = 0.99
17
+ Discrete.clip_prob_low = 0.001
18
+
19
+ # Parameters for Experiment:
20
+ # ==============================================================================
21
+ Experiment.always_load_latest = False
22
+ Experiment.always_save_latest = True
23
+ Experiment.has_dset_edit_rights = True
24
+ Experiment.save_trajs_as = 'npz'
25
+ Experiment.stagger_traj_file_lengths = True
26
+ Experiment.traj_save_len = 10000000000.0
27
+ Experiment.wandb_group_name = None
28
+
29
+ # Parameters for FlashAttention:
30
+ # ==============================================================================
31
+ FlashAttention.window_size = (32, 0)
32
+
33
+ # Parameters for MetamonAMAGOExperiment:
34
+ # ==============================================================================
35
+ MetamonAMAGOExperiment.agent_type = @agent.MultiTaskAgent
36
+ MetamonAMAGOExperiment.critic_loss_weight = 10.0
37
+ MetamonAMAGOExperiment.grad_clip = 1.5
38
+ MetamonAMAGOExperiment.l2_coeff = 0.0001
39
+ MetamonAMAGOExperiment.learning_rate = 0.00015
40
+ MetamonAMAGOExperiment.lr_warmup_steps = 1000
41
+ MetamonAMAGOExperiment.max_seq_len = 200
42
+ MetamonAMAGOExperiment.traj_encoder_type = @traj_encoders.TformerTrajEncoder
43
+ MetamonAMAGOExperiment.tstep_encoder_type = @MetamonTstepEncoder
44
+
45
+ # Parameters for MetamonMaskedActor:
46
+ # ==============================================================================
47
+ MetamonMaskedActor.activation = 'leaky_relu'
48
+ MetamonMaskedActor.continuous_dist_type = None
49
+ MetamonMaskedActor.d_hidden = 400
50
+ MetamonMaskedActor.dropout_p = 0.0
51
+ MetamonMaskedActor.mask_illegal_actions = True
52
+ MetamonMaskedActor.n_layers = 2
53
+
54
+ # Parameters for MetamonTstepEncoder:
55
+ # ==============================================================================
56
+ MetamonTstepEncoder.d_model = 100
57
+ MetamonTstepEncoder.dropout = 0.05
58
+ MetamonTstepEncoder.extra_emb_dim = 18
59
+ MetamonTstepEncoder.n_heads = 5
60
+ MetamonTstepEncoder.n_layers = 3
61
+ MetamonTstepEncoder.numerical_tokens = 6
62
+ MetamonTstepEncoder.scratch_tokens = 6
63
+ MetamonTstepEncoder.token_mask_aug = False
64
+
65
+ # Parameters for Multigammas:
66
+ # ==============================================================================
67
+ Multigammas.continuous = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
68
+ Multigammas.discrete = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
69
+
70
+ # Parameters for MultiTaskAgent:
71
+ # ==============================================================================
72
+ MultiTaskAgent.actor_type = @MetamonMaskedActor
73
+ MultiTaskAgent.critic_type = @actor_critic.NCriticsTwoHot
74
+ MultiTaskAgent.fake_filter = False
75
+ MultiTaskAgent.fbc_filter_func = @agent.binary_filter
76
+ MultiTaskAgent.gamma = 0.999
77
+ MultiTaskAgent.num_actions_for_value_in_actor_loss = 3
78
+ MultiTaskAgent.num_actions_for_value_in_critic_loss = 5
79
+ MultiTaskAgent.num_critics = 6
80
+ MultiTaskAgent.num_critics_td = 2
81
+ MultiTaskAgent.offline_coeff = 1.0
82
+ MultiTaskAgent.online_coeff = 0.0
83
+ MultiTaskAgent.pass_obs_keys_to_actor = ['illegal_actions']
84
+ MultiTaskAgent.popart = True
85
+ MultiTaskAgent.reward_multiplier = 10.0
86
+ MultiTaskAgent.tau = 0.004
87
+ MultiTaskAgent.use_multigamma = True
88
+ MultiTaskAgent.use_target_actor = True
89
+
90
+ # Parameters for NCriticsTwoHot:
91
+ # ==============================================================================
92
+ NCriticsTwoHot.activation = 'leaky_relu'
93
+ NCriticsTwoHot.d_hidden = 512
94
+ NCriticsTwoHot.dropout_p = 0.0
95
+ NCriticsTwoHot.max_return = 1100
96
+ NCriticsTwoHot.min_return = -1100
97
+ NCriticsTwoHot.n_layers = 2
98
+ NCriticsTwoHot.output_bins = 96
99
+ NCriticsTwoHot.use_symlog = False
100
+
101
+ # Parameters for PopArtLayer:
102
+ # ==============================================================================
103
+ PopArtLayer.beta = 0.0005
104
+ PopArtLayer.init_nu = 100.0
105
+
106
+ # Parameters for SigmaReparam:
107
+ # ==============================================================================
108
+ SigmaReparam.bias = True
109
+ SigmaReparam.fast_init = False
110
+
111
+ # Parameters for TformerTrajEncoder:
112
+ # ==============================================================================
113
+ TformerTrajEncoder.activation = 'leaky_relu'
114
+ TformerTrajEncoder.d_ff = 3072
115
+ TformerTrajEncoder.d_model = 768
116
+ TformerTrajEncoder.dropout_attn = 0.0
117
+ TformerTrajEncoder.dropout_emb = 0.05
118
+ TformerTrajEncoder.dropout_ff = 0.05
119
+ TformerTrajEncoder.dropout_qkv = 0.0
120
+ TformerTrajEncoder.head_scaling = True
121
+ TformerTrajEncoder.n_heads = 8
122
+ TformerTrajEncoder.n_layers = 6
123
+ TformerTrajEncoder.norm = 'layer'
124
+ TformerTrajEncoder.normformer_norms = True
125
+ TformerTrajEncoder.pos_emb = 'fixed'
126
+ TformerTrajEncoder.sigma_reparam = True
127
+
128
+ # Parameters for TransformerTurnEmbedding:
129
+ # ==============================================================================
130
+ # None.
abra/ckpts/latest/policy.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:294f289899db013382b6e781ffe30f1176086a26a18b94def13f396bbdd77cc9
3
+ size 230529797
abra/ckpts/policy_weights/policy_epoch_0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:799132537bd8eb385daaa0043b967715eabd4b51919296aef6cadf2c9ff09774
3
+ size 230535701
abra/ckpts/policy_weights/policy_epoch_10.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be8fd88ccd9ef46b179ba55a9da44bf43557822616e7407b30dac1f9efd7684e
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_12.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:935cf751c557146544e6ebaa53640ac1f053fffca906a9dba3259cec7dbd7617
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_14.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fa0f48a1121ce9258ff4fba3290d0b2726452ee88a53b1edf38f316b4ec9c09
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_16.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74d42393f3365ecb0024cf993dac2ccc1ebd953fa4cd01f059a9ebc523438222
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_18.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c710c2f234de461388715993215c7f05358ac01cb4d66d2c9008501d060eac31
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86f8ed5119c324db913793dedbf46bdacc58ea760db671a6737236a43d241016
3
+ size 230535701
abra/ckpts/policy_weights/policy_epoch_20.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5be29d8b23eccc0d6406c483c9a723c18710a05da000fda0259e3ba205dca27
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_22.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd825e560f4b862165ed3d268918557f6d2ed224115b506710f28a855b814643
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_24.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0e97e454487fd82a985c1f2bc8a93bcb6c0827e4f7600cdad44f81f7376524f
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_26.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10559b27a2bad809a843623cd6a11d6161cd10631713c1663160e243f3ab7aef
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_28.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f0f8672a03aae22abfe7d5d3266a38dd537f5ec01c1649ec289908d4187ef9d
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_30.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:776d2b2aae2fddb5f1bf336e23ad616ff97e16fc1d3bad959f531ab1c5ed341e
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_32.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34a17918a18ac5dd21d21ff99e7e0eb3f1811264971263da0283862481486150
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_34.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8310f2e7cb2eafbe1588d312a22670513c8d43387aeb836669c6c31341a39a2f
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_36.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c1911570f6f970c3135f370bc8fcc1bd0cc8c8ce0fbbc26a90c66dc17fafb90
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_38.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f9f49e880cef3e3b0b4717f93f7599bc0a83aeea5e9ede2757b8eb41eaa4f2e
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7aeb4c44e7dfa55b590fb495c44609bc9f736869b8438c1afce4d90c8f7f7042
3
+ size 230535701
abra/ckpts/policy_weights/policy_epoch_40.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f473d8b5cfb344925b753e0477b327c81012c01d000aa98e31096e25479b895
3
+ size 230535959
abra/ckpts/policy_weights/policy_epoch_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f68878738b3ac3eb13dd8a9b9b2ad05326850976c275abf225022e0ec8cddfa8
3
+ size 230535701
abra/ckpts/policy_weights/policy_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9bb3aa2162ff1f950d652f2ea13ea34613cdfd51c32a105d892546e62fa8f55c
3
+ size 230535701
abra/ckpts/training_states/gen9v3_epoch_0/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5927ba6b5ec6da64a498e5354cb8a241d0dcc3f9573f79cd59af098e2f2bbe3
3
+ size 1461
abra/ckpts/training_states/gen9v3_epoch_0/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a62867c59a2c936134763270de1d85fc33a5c767eb4a69c682a2811555f6d7ce
3
+ size 230466404
abra/ckpts/training_states/gen9v3_epoch_0/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c17616a7f3c0608bca7cee769a8fbd8dde283945b48818fa3f69c059c1c4dfc
3
+ size 387965643
abra/ckpts/training_states/gen9v3_epoch_0/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05c38ddd011dab36ce21e6495921272fb835715d5e6c8c1caf848179afe74e8d
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_0/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6971a36466b5658922262c8c02ff0d3e548a03ffd3970090253e526b9e88646e
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_0/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4137f674c1b2ee9634bcc25cd8ca594d21349c0d7f22bb5dae52be1b3ef4f1e1
3
+ size 15277
abra/ckpts/training_states/gen9v3_epoch_0/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5093b54b40e8f5b0287e96ba7dbe0c652715dcc331a77d7940c917497d8a32c
3
+ size 1401
abra/ckpts/training_states/gen9v3_epoch_10/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3fc725bfcbdbea33e89a28fab6a79d73eef12c38dc575979b76e1a980b0f351
3
+ size 1461
abra/ckpts/training_states/gen9v3_epoch_10/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f6f635899553110fc7064b8c1cfef958bdb9615f1db4aa778bd451a8e070444
3
+ size 230466404
abra/ckpts/training_states/gen9v3_epoch_10/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6ec73e3bbe78cbd86902129cb09d6547e7d384bf8bc69fbe945e3239a5e76f7
3
+ size 387965643
abra/ckpts/training_states/gen9v3_epoch_10/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8251fe6eb5a0ed8783b55f520d09fad7fd34e50f1787e4ce086ddd0ccae84b2c
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_10/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91243974d25aa369b6aec7c879a28bfae39470b3c854920645d71872b5e58410
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_10/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b82e28106256da527d6049685ad5ceeb602d0adec0292f1a940e646fe8930d4
3
+ size 15277
abra/ckpts/training_states/gen9v3_epoch_10/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5e234b082cb41a9ff43bc6b11fe71f55e543ad78151e283fd7a7f5567d6ae92
3
+ size 1401
abra/ckpts/training_states/gen9v3_epoch_12/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae64af5bcea651253e06c2189ff3bddbe30ee6fc93d82f4957caef0d813ed446
3
+ size 1461
abra/ckpts/training_states/gen9v3_epoch_12/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1a08890bed24624488b67ac4ca7acdf0f70369680f64aa995ead44fca08c0bd
3
+ size 230466404
abra/ckpts/training_states/gen9v3_epoch_12/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40ce529f2509397afd588d597c83fc23eb4ad02ad2f5cf1b9e92f1962308795f
3
+ size 387965643
abra/ckpts/training_states/gen9v3_epoch_12/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce5751ce107aec55112132bddbc1bcc70c618dcc33d30604de0be7e19dadf4fa
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_12/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d8d38587733cda6254fe9da135aa3ff00c8fd133b5fd99283ecb94e55affb9a
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_12/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c1d06030c42a2a08ebefe688305a55c1cc5dc5a79dd8661d604e2c57c618712
3
+ size 15277
abra/ckpts/training_states/gen9v3_epoch_12/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ea4421de7eec1cb41eaf9961084f423cd1a3a24b5a5fcb58deec98f16a27795
3
+ size 1401
abra/ckpts/training_states/gen9v3_epoch_14/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fba5418c5e65ea8e19ff57c5f68e0c7a8e1b0b607310cbe47c96ed11fe49f4e6
3
+ size 1461
abra/ckpts/training_states/gen9v3_epoch_14/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:739a99e95453019f42c1eadc0bb18a3b14e42b0aec53f0594aa3cc11ec6f68e0
3
+ size 230466404
abra/ckpts/training_states/gen9v3_epoch_14/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b746d2038228defb1e49dff116f9ae43ceb5da77e2169a0571c058b56caff33
3
+ size 387965643
abra/ckpts/training_states/gen9v3_epoch_14/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01e96398a1b32c79e7d212df818356c0cd2ae85de029ec1ab2b53e4dcc0ce391
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_14/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e912a5977d567bb504d0adccb3d868e8ddabe73b0fd092c6e335bd2ca934f515
3
+ size 15341
abra/ckpts/training_states/gen9v3_epoch_14/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b069b3e675fbe7d54dc649efb780ce1e4bbb1739b4bcd4c1a13a067c9fd44850
3
+ size 15277