jw2yang commited on
Commit
1b73edf
1 Parent(s): 80a17c3

Update detectron2/config/defaults.py

Browse files
Files changed (1) hide show
  1. detectron2/config/defaults.py +2 -2
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 = "cuda"
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