update

DefangChen · Oct 22, 2019 · e5fbb28 · e5fbb28
1 parent c513466
commit e5fbb28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_student.py b/train_student.py
@@ -283,7 +283,7 @@ def main():
  cudnn.benchmark = True
 
  # validate teacher accuracy
- teacher_acc, _ = validate(val_loader, model_t, criterion_cls, opt)
+ teacher_acc, _, _ = validate(val_loader, model_t, criterion_cls, opt)
  print('teacher accuracy: ', teacher_acc)
 
  # routine