joseph-tennyson commited on
Commit
806cf90
·
verified ·
1 Parent(s): 6a47351

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. config.yaml +44 -0
  2. model_4000.pt +3 -0
  3. model_8000.pt +3 -0
  4. 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