add comments

DefangChen · Jan 11, 2020 · 7fd6eff · 7fd6eff
1 parent 102332d
commit 7fd6eff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/train_student.py b/train_student.py
@@ -330,6 +330,8 @@ def main():
  save_file = os.path.join(opt.save_folder, 'ckpt_epoch_{epoch}.pth'.format(epoch=epoch))
  torch.save(state, save_file)
 
+ # This best accuracy is only for printing purpose.
+ # The results reported in the paper/README is from the last epoch. 
  print('best accuracy:', best_acc)
 
  # save model

diff --git a/train_teacher.py b/train_teacher.py
@@ -158,6 +158,8 @@ def main():
  save_file = os.path.join(opt.save_folder, 'ckpt_epoch_{epoch}.pth'.format(epoch=epoch))
  torch.save(state, save_file)
 
+ # This best accuracy is only for printing purpose.
+ # The results reported in the paper/README is from the last epoch.
  print('best accuracy:', best_acc)
 
  # save model