tharun2909 commited on
Commit
e773134
·
verified ·
1 Parent(s): a491626

Upload training configuration

Browse files
Files changed (1) hide show
  1. run_config.json +18 -0
run_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "distilgpt2",
3
+ "epochs": 2,
4
+ "batch_size": 1,
5
+ "learning_rate": 0.0003,
6
+ "grad_accum_steps": 4,
7
+ "precision": "fp32",
8
+ "quantization_bits": 8,
9
+ "lora_r": 4,
10
+ "lora_alpha": 8,
11
+ "lora_dropout": 0.04,
12
+ "target_modules": [
13
+ "c_attn",
14
+ "c_proj"
15
+ ],
16
+ "total_steps": 60,
17
+ "num_train_samples": 240
18
+ }