update

yolo/models/yolov7_af/yolov7_af_backbone.py

@@ -8,7 +8,7 @@ except:
 
 # IN1K pretrained weight
 pretrained_urls = {
-    't': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elannet_t_in1k_62.0.pth",
+    't': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elannet_t_in1k_63.2.pth",
     'l': None,
     'x': None,
 }

yolo/train.py

@@ -196,7 +196,7 @@ def train():
         trainer.eval(model_eval)
         return
 
-    garbage = torch.randn(640, 1024, 73, 73).to(device) # 15 G
+    # garbage = torch.randn(640, 1024, 73, 73).to(device) # 15 G
 
     # ---------------------------- Train pipeline ----------------------------
     trainer.train(model)