Ge Zheng
commited on
Commit
·
e3b4b18
1
Parent(s):
2f486a3
Update train.py
Browse files- tools/train.py +3 -3
tools/train.py
CHANGED
@@ -63,11 +63,11 @@ def make_parser():
|
|
63 |
)
|
64 |
parser.add_argument(
|
65 |
"-o",
|
66 |
-
"--
|
67 |
-
dest="
|
68 |
default=False,
|
69 |
action="store_true",
|
70 |
-
help="
|
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",
|