train YOLOv1-R18

models/yolov1/yolov1_pred.py

@@ -1,7 +1,5 @@
-import math
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
 
 
 # -------------------- Detection Pred Layer --------------------