float16/float32 are supported

apache · Aug 1, 2019 · 7da713b · 7da713b
1 parent 0decf31
commit 7da713b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/example/rnn/bucketing/cudnn_rnn_bucketing.py b/example/rnn/bucketing/cudnn_rnn_bucketing.py
@@ -69,7 +69,8 @@
 parser.add_argument('--rnntype', type=str, default='lstm',
                     help='rnn type: gru, lstm, rnn_tanh and rnn_relu are supported')
 parser.add_argument('--dtype', type=str, default='float32',
-                    help='if float16 is provided AMP convert model'
+                    help='float16 and float32 are supported'
+                         'if float16 is provided AMP convert model'
                          'is used to convert model to mixed precision model'
                          'before running inference')