Spaces:

HemaAM
/

YoloV3_PascalVOC_Dataset

Sleeping

HemaAM commited on Aug 25, 2023

Commit

d4fb5fa

1 Parent(s): 5c5fbf4

Bug Fix for use_cuda

Files changed (1) hide show

detection.py CHANGED Viewed

@@ -20,7 +20,7 @@ scaled_anchors = (
     * torch.tensor(modelConfig.S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2)
 ).to(modelConfig.DEVICE)
-yolo_grad_cam = YoloGradCAM(model=model, target_layers=[model.layers[-2]], use_cuda=True)
 @torch.inference_mode()

     * torch.tensor(modelConfig.S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2)
 ).to(modelConfig.DEVICE)
+yolo_grad_cam = YoloGradCAM(model=model, target_layers=[model.layers[-2]], use_cuda=False)
 @torch.inference_mode()