Spaces:

TraceForce
/

varun-kd-finetune

Sleeping

Varun Wadhwa commited on about 1 month ago

Commit

04171cd

unverified ·

1 Parent(s): 1f7e238

Logs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,7 @@ def distillation_loss(student_logits, teacher_logits, true_labels, temperature,
 # hyperparameters
 batch_size = 32
 lr = 1e-4
-num_epochs = 15
 temperature = 2.0
 alpha = 0.5

 # hyperparameters
 batch_size = 32
 lr = 1e-4
+num_epochs = 10
 temperature = 2.0
 alpha = 0.5