Spaces:
Runtime error
Runtime error
Update detectron2/config/defaults.py
Browse files
detectron2/config/defaults.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
# Copyright (c) Facebook, Inc. and its affiliates.
|
2 |
from .config import CfgNode as CN
|
3 |
|
4 |
-
#
|
5 |
# Convention about Training / Test specific parameters
|
6 |
# -----------------------------------------------------------------------------
|
7 |
# Whenever an argument can be either used for training or for testing, the
|
@@ -25,7 +25,7 @@ _C.MODEL = CN()
|
|
25 |
_C.MODEL.LOAD_PROPOSALS = False
|
26 |
_C.MODEL.MASK_ON = False
|
27 |
_C.MODEL.KEYPOINT_ON = False
|
28 |
-
_C.MODEL.DEVICE = "
|
29 |
_C.MODEL.META_ARCHITECTURE = "GeneralizedRCNN"
|
30 |
|
31 |
# Path (a file path, or URL like detectron2://.., https://..) to a checkpoint file
|
|
|
1 |
# Copyright (c) Facebook, Inc. and its affiliates.
|
2 |
from .config import CfgNode as CN
|
3 |
|
4 |
+
# ------------------------------------------------------------------de-----------
|
5 |
# Convention about Training / Test specific parameters
|
6 |
# -----------------------------------------------------------------------------
|
7 |
# Whenever an argument can be either used for training or for testing, the
|
|
|
25 |
_C.MODEL.LOAD_PROPOSALS = False
|
26 |
_C.MODEL.MASK_ON = False
|
27 |
_C.MODEL.KEYPOINT_ON = False
|
28 |
+
_C.MODEL.DEVICE = "cpu"
|
29 |
_C.MODEL.META_ARCHITECTURE = "GeneralizedRCNN"
|
30 |
|
31 |
# Path (a file path, or URL like detectron2://.., https://..) to a checkpoint file
|