AMP training level no more than O1

Signed-off-by: begeekmyfriend <[email protected]>
begeekmyfriend · Feb 25, 2020 · 57dddb1 · 57dddb1
1 parent bbdd596
commit 57dddb1
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/hparams.py b/hparams.py
@@ -12,7 +12,6 @@
 # set this to True if you are only interested in WaveRNN
 ignore_tts = True
 amp = True
-amp_level = 'O2'
 
 
 # DSP --------------------------------------------------------------------------------------------------------------#

diff --git a/models/fatchord_version.py b/models/fatchord_version.py
@@ -229,7 +229,7 @@ def generate(self, mels, save_path, batched, target, overlap, mu_law):
             output = output[0]
 
         end = time.time()
-        print(f'{end - start}')
+        print(f'Elapsed {end - start} seconds')
         return save_wav(output[:wave_len], save_path)
 
 

diff --git a/train_wavernn.py b/train_wavernn.py
@@ -33,7 +33,7 @@ def voc_train_loop(model, loss_func, optimizer, train_set, test_set, init_lr, fi
     epochs = int((total_steps - model.get_step()) // total_iters + 1)
 
     if hp.amp:
-        model, optimizer = amp.initialize(model, optimizer, opt_level=hp.amp_level)
+        model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
 
     torch.backends.cudnn.benchmark = True