Lung_Nodule_Detection_Dicom

Runtime error

basilshaji commited on Jul 14, 2024

Commit

37c35d9

verified ·

1 Parent(s): 7bc8c1c

Update yolov5/utils/autobatch.py

Files changed (1) hide show

yolov5/utils/autobatch.py CHANGED Viewed

@@ -6,8 +6,8 @@ from copy import deepcopy
 import numpy as np
 import torch
-from utils.general import LOGGER, colorstr
-from utils.torch_utils import profile
 def check_train_batch_size(model, imgsz=640, amp=True):

 import numpy as np
 import torch
+from yolov5.utils.general import LOGGER, colorstr
+from yolov5.utils.torch_utils import profile
 def check_train_batch_size(model, imgsz=640, amp=True):