Add visualize codes for Q

2021-04-11 21:45:20 +08:00
parent e777f38233
commit 0e2dd13762
16 changed files with 570 additions and 125 deletions
--- a/lib/procedures/simple_KD_main.py
+++ b/lib/procedures/simple_KD_main.py
@@ -4,9 +4,9 @@
 import os, sys, time, torch
 import torch.nn.functional as F

-# our modules
+# modules in AutoDL
 from log_utils import AverageMeter, time_string
-from utils import obtain_accuracy
+from .eval_funcs import obtain_accuracy


 def simple_KD_train(