Jonas Steinmann
commited on
Commit
·
c9cdfa7
1
Parent(s):
a8ad894
Fix --fp16 option (#247)
Browse filesDefault needs to be set to False, for the option to have any effect at all
- tools/train.py +1 -1
tools/train.py
CHANGED
@@ -64,7 +64,7 @@ def make_parser():
|
|
64 |
parser.add_argument(
|
65 |
"--fp16",
|
66 |
dest="fp16",
|
67 |
-
default=
|
68 |
action="store_true",
|
69 |
help="Adopting mix precision training.",
|
70 |
)
|
|
|
64 |
parser.add_argument(
|
65 |
"--fp16",
|
66 |
dest="fp16",
|
67 |
+
default=False,
|
68 |
action="store_true",
|
69 |
help="Adopting mix precision training.",
|
70 |
)
|