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
|