Fixed comment on GPU_COUNT (matterport#878)

Fixed comment on GPU_COUNT
kunthet · Sep 21, 2018 · 80cac4c · 80cac4c
1 parent 61834b4
commit 80cac4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mrcnn/config.py b/mrcnn/config.py
@@ -24,7 +24,7 @@ class Config(object):
     # experiment is running.
     NAME = None  # Override in sub-classes
 
-    # NUMBER OF GPUs to use. For CPU training, use 1
+    # NUMBER OF GPUs to use. When using only a CPU, this needs to be set to 1.
     GPU_COUNT = 1
 
     # Number of images to train with on each GPU. A 12GB GPU can typically