Upload folder using huggingface_hub
Browse files- config.yaml +44 -0
- model_4000.pt +3 -0
- model_8000.pt +3 -0
- state.pt +3 -0
config.yaml
ADDED
|
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
config: /root/in-context-learning-GLM/src/glm_configs/joint_0.32_5.yaml
|
| 2 |
+
model:
|
| 3 |
+
family: gpt2
|
| 4 |
+
n_dims: 10
|
| 5 |
+
n_embd: 256
|
| 6 |
+
n_head: 8
|
| 7 |
+
n_layer: 12
|
| 8 |
+
n_positions: 40
|
| 9 |
+
out_dir: /root/in-context-learning-GLM/models/joint-0.32-5.0
|
| 10 |
+
test_run: false
|
| 11 |
+
training:
|
| 12 |
+
batch_size: 256
|
| 13 |
+
curriculum:
|
| 14 |
+
dims:
|
| 15 |
+
end: 10
|
| 16 |
+
inc: 0
|
| 17 |
+
interval: 10000
|
| 18 |
+
start: 10
|
| 19 |
+
points:
|
| 20 |
+
end: 40
|
| 21 |
+
inc: 0
|
| 22 |
+
interval: 10000
|
| 23 |
+
start: 40
|
| 24 |
+
data: gaussian
|
| 25 |
+
keep_every_steps: 4000
|
| 26 |
+
learning_rate: 0.00025
|
| 27 |
+
num_tasks: null
|
| 28 |
+
num_training_examples: null
|
| 29 |
+
resume_id: joint-0.32-5.0
|
| 30 |
+
save_every_steps: 4000
|
| 31 |
+
task: GLM
|
| 32 |
+
task_kwargs:
|
| 33 |
+
function_type:
|
| 34 |
+
- poisson
|
| 35 |
+
- neg_binomial
|
| 36 |
+
r: 5.0
|
| 37 |
+
scaling: 0.32
|
| 38 |
+
train_steps: 20000
|
| 39 |
+
wandb:
|
| 40 |
+
entity: in-context
|
| 41 |
+
log_every_steps: 10
|
| 42 |
+
name: joint-0.32-5.0
|
| 43 |
+
notes: ICL GLM training
|
| 44 |
+
project: in-context-training
|
model_4000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c57b02e4ad4b21b0c9ab4cb667056559c9c38ea9757e389ef8e625c40e62552c
|
| 3 |
+
size 89608311
|
model_8000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:79560f90394dcebfa0120ed5f80e4a2f539727798abc6e58bc868c657881d70d
|
| 3 |
+
size 89608311
|
state.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:89274a60bbf987d09434fb977e028cf32a8f26f66d856d7371566227984ad2c3
|
| 3 |
+
size 165713513
|