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/search_main.py
+++ b/lib/procedures/search_main.py
@@ -3,9 +3,10 @@
 ##################################################
 import os, sys, time, torch
 from log_utils import AverageMeter, time_string
-from utils import obtain_accuracy
 from models import change_key

+from .eval_funcs import obtain_accuracy
+

 def get_flop_loss(expected_flop, flop_cur, flop_need, flop_tolerant):
    expected_flop = torch.mean(expected_flop)