Update src/run/yolov3/config.py
Browse files- src/run/yolov3/config.py +1 -1
src/run/yolov3/config.py
CHANGED
@@ -3,7 +3,7 @@ import cv2
|
|
3 |
import torch
|
4 |
|
5 |
from albumentations.pytorch import ToTensorV2
|
6 |
-
|
7 |
|
8 |
DATASET = "PASCAL_VOC"
|
9 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
3 |
import torch
|
4 |
|
5 |
from albumentations.pytorch import ToTensorV2
|
6 |
+
|
7 |
|
8 |
DATASET = "PASCAL_VOC"
|
9 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|