Jonas Steinmann commited on
Commit
c9cdfa7
·
1 Parent(s): a8ad894

Fix --fp16 option (#247)

Browse files

Default needs to be set to False, for the option to have any effect at all

Files changed (1) hide show
  1. 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=True,
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
  )