File size: 1,364 Bytes
a2edaba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
{
  "teacher_model": "gpt2-large",
  "student_model": "distilgpt2",
  "tokenizer_model": "gpt2",
  "batch_size": 4,
  "max_length": 256,
  "num_epochs": 1,
  "learning_rate": 5e-05,
  "distillation_alpha": 0.7,
  "student_lm_beta": 0.3,
  "dataset_file": "/home/jayanthram.k/Distillation/clean_dataset_with_no_outliers.csv",
  "train_samples": 43572,
  "val_samples": 4842,
  "eval_samples": 100,
  "best_validation_loss": 2.3524647863515638,
  "evaluation_metrics": {
    "rouge1_teacher_student_before": 0.32189833105454313,
    "rouge2_teacher_student_before": 0.07253675599508494,
    "rougeL_teacher_student_before": 0.19562937213425652,
    "cosine_similarity_teacher_student_before": 0.44240760803222656,
    "rouge1_teacher_student_after": 0.3346180933711459,
    "rouge2_teacher_student_after": 0.08527329173516719,
    "rougeL_teacher_student_after": 0.20919089238473532,
    "cosine_similarity_teacher_student_after": 0.5256437659263611,
    "rouge1_student_before_student_after": 0.3137800181943293,
    "rouge2_student_before_student_after": 0.07712439299277635,
    "rougeL_student_before_student_after": 0.18683832802096778,
    "cosine_similarity_student_before_student_after": 0.4463849663734436,
    "avg_teacher_time": 1.2837520718574524,
    "avg_student_before_time": 0.4636122870445252,
    "avg_student_after_time": 0.46685707569122314
  }
}