alishudi commited on
Commit
6fbe18c
1 Parent(s): 7e30e3f

Upload parameters.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. parameters.json +52 -0
parameters.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "force": true,
3
+ "dump_path": "./serialization_dir/distilbert",
4
+ "data_file": "data/binarized_text.bert.pickle",
5
+ "student_type": "distilbert",
6
+ "student_config": "training_configs/distilbert-base-uncased.json",
7
+ "student_pretrained_weights": "./serialization_dir/tf_bert-base-uncased_0247911.pth",
8
+ "teacher_type": "bert",
9
+ "teacher_name": "bert-base-uncased",
10
+ "temperature": 2.0,
11
+ "alpha_ce": 5.0,
12
+ "alpha_mlm": 2.0,
13
+ "alpha_clm": 0.0,
14
+ "alpha_mse": 0.0,
15
+ "alpha_cos": 1.0,
16
+ "alpha_act": 1.0,
17
+ "mlm": true,
18
+ "mlm_mask_prop": 0.15,
19
+ "word_mask": 0.8,
20
+ "word_keep": 0.1,
21
+ "word_rand": 0.1,
22
+ "mlm_smoothing": 0.7,
23
+ "token_counts": "data/token_counts.bert.pickle",
24
+ "restrict_ce_to_mask": false,
25
+ "freeze_pos_embs": true,
26
+ "freeze_token_type_embds": false,
27
+ "n_epoch": 50,
28
+ "batch_size": 5,
29
+ "group_by_size": true,
30
+ "gradient_accumulation_steps": 50,
31
+ "warmup_prop": 0.05,
32
+ "weight_decay": 0.0,
33
+ "learning_rate": 0.0005,
34
+ "adam_epsilon": 1e-06,
35
+ "max_grad_norm": 5.0,
36
+ "initializer_range": 0.02,
37
+ "fp16": false,
38
+ "fp16_opt_level": "O1",
39
+ "n_gpu": 1,
40
+ "local_rank": 0,
41
+ "seed": 56,
42
+ "log_interval": 500,
43
+ "checkpoint_interval": 4000,
44
+ "n_nodes": 1,
45
+ "node_id": 0,
46
+ "global_rank": 0,
47
+ "world_size": 1,
48
+ "n_gpu_per_node": 1,
49
+ "multi_gpu": false,
50
+ "is_master": true,
51
+ "multi_node": false
52
+ }