瀏覽代碼

remove grad accumulate

yjh0410 1 年之前
父節點
當前提交
e3f68f196d
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      yolo/engine.py

+ 1 - 1
yolo/engine.py

@@ -6,7 +6,7 @@ import random
 
 
 # ----------------- Extra Components -----------------
 # ----------------- Extra Components -----------------
 from utils import distributed_utils
 from utils import distributed_utils
-from utils.misc import MetricLogger, SmoothedValue, get_total_grad_norm
+from utils.misc import MetricLogger, SmoothedValue
 from utils.vis_tools import vis_data
 from utils.vis_tools import vis_data
 
 
 # ----------------- Optimizer & LrScheduler Components -----------------
 # ----------------- Optimizer & LrScheduler Components -----------------