set argument '--gpus' to '0'

liyangdev · Jan 4, 2016 · 1e5e8c0 · 1e5e8c0
1 parent 0b66c47
commit 1e5e8c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tools/accnn/acc_conv.py b/tools/accnn/acc_conv.py
@@ -67,11 +67,11 @@ def main():
 if __name__ == '__main__':
   parser=argparse.ArgumentParser()
   parser.add_argument('-m', '--model', help='the model to speed up')
-  parser.add_argument('-g', '--gpus', default='0,1,2,3', help='the gpus to be used in ctx')
+  parser.add_argument('-g', '--gpus', default='0', help='the gpus to be used in ctx')
   parser.add_argument('--load-epoch',type=int,default=1)
   parser.add_argument('--layer')
   parser.add_argument('--K', type=int)
   parser.add_argument('--save-model')
   args = parser.parse_args()
   main()
-
+
diff --git a/tools/accnn/acc_fc.py b/tools/accnn/acc_fc.py
@@ -48,7 +48,7 @@ def main():
 if __name__ == '__main__':
   parser=argparse.ArgumentParser()
   parser.add_argument('-m', '--model', help='the model to speed up')
-  parser.add_argument('-g', '--gpus', default='0,1,2,3', help='the gpus to be used in ctx')
+  parser.add_argument('-g', '--gpus', default='0', help='the gpus to be used in ctx')
   parser.add_argument('--load-epoch',type=int,default=1)
   parser.add_argument('--layer')
   parser.add_argument('--K', type=int)