Ge Zheng commited on
Commit
e3b4b18
·
1 Parent(s): 2f486a3

Update train.py

Browse files
Files changed (1) hide show
  1. tools/train.py +3 -3
tools/train.py CHANGED
@@ -63,11 +63,11 @@ def make_parser():
63
  )
64
  parser.add_argument(
65
  "-o",
66
- "--occumpy",
67
- dest="occumpy",
68
  default=False,
69
  action="store_true",
70
- help="occumpy GPU memory first for training.",
71
  )
72
  parser.add_argument(
73
  "opts",
 
63
  )
64
  parser.add_argument(
65
  "-o",
66
+ "--occupy",
67
+ dest="occupy",
68
  default=False,
69
  action="store_true",
70
+ help="occupy GPU memory first for training.",
71
  )
72
  parser.add_argument(
73
  "opts",