From e0bd1423b526ca30c11368d647b098d39ca4f979 Mon Sep 17 00:00:00 2001
From: Vinh Nguyen <vinhn@nvidia.com>
Date: Mon, 14 Oct 2019 12:34:29 +0000
Subject: [PATCH] adding automatic mixed precision training support

---
 util/flags.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/util/flags.py b/util/flags.py
index 80251cb0..2e585a7c 100644
--- a/util/flags.py
+++ b/util/flags.py
@@ -80,6 +80,8 @@ def create_flags():
     f.DEFINE_boolean('use_cudnn_rnn', False, 'use CuDNN RNN backend for training on GPU. Note that checkpoints created with this flag can only be used with CuDNN RNN, i.e. fine tuning on a CPU device will not work')
     f.DEFINE_string('cudnn_checkpoint', '', 'path to a checkpoint created using --use_cudnn_rnn. Specifying this flag allows one to convert a CuDNN RNN checkpoint to a checkpoint capable of running on a CPU graph.')
 
+    f.DEFINE_boolean('automatic_mixed_precision', False, 'whether to allow automatic mixed precision training')
+    
     # Sample limits
 
     f.DEFINE_integer('limit_train', 0, 'maximum number of elements to use from train set - 0 means no limit')