[TTS] enforced pin_memory = True (NVIDIA#4341)

Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com> Signed-off-by: stevehuang52 <heh@nvidia.com>
stevehuang52 · Jun 8, 2022 · 770f259 · 770f259
1 parent 77970cc
commit 770f259
Show file tree

Hide file tree

Showing 16 changed files with 24 additions and 8 deletions.
diff --git a/examples/tts/conf/aligner.yaml b/examples/tts/conf/aligner.yaml
@@ -84,7 +84,7 @@ model:
       shuffle: true
       batch_size: 64
       num_workers: 4
-      pin_memory: false
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -110,7 +110,7 @@ model:
       shuffle: false
       batch_size: 64
       num_workers: 1
-      pin_memory: false
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor

diff --git a/examples/tts/conf/fastpitch_align_44100.yaml b/examples/tts/conf/fastpitch_align_44100.yaml
@@ -108,6 +108,7 @@ model:
       shuffle: true
       batch_size: 32
       num_workers: 12
+      pin_memory: true
 
 
   validation_ds:
@@ -140,6 +141,7 @@ model:
       shuffle: false
       batch_size: 32
       num_workers: 8
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor

diff --git a/examples/tts/conf/fastpitch_align_v1.05.yaml b/examples/tts/conf/fastpitch_align_v1.05.yaml
@@ -109,6 +109,7 @@ model:
       shuffle: true
       batch_size: 32
       num_workers: 12
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -140,6 +141,7 @@ model:
       shuffle: false
       batch_size: 32
       num_workers: 8
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor

diff --git a/examples/tts/conf/hifigan/model/train_ds/train_ds.yaml b/examples/tts/conf/hifigan/model/train_ds/train_ds.yaml
@@ -10,3 +10,4 @@ dataloader_params:
   shuffle: true
   batch_size: 16
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/hifigan/model/train_ds/train_ds_finetune.yaml b/examples/tts/conf/hifigan/model/train_ds/train_ds_finetune.yaml
@@ -12,3 +12,4 @@ dataloader_params:
   shuffle: true
   batch_size: 16
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/hifigan/model/validation_ds/val_ds.yaml b/examples/tts/conf/hifigan/model/validation_ds/val_ds.yaml
@@ -10,3 +10,4 @@ dataloader_params:
   shuffle: false
   batch_size: 16
   num_workers: 1
+  pin_memory: true
diff --git a/examples/tts/conf/hifigan/model/validation_ds/val_ds_finetune.yaml b/examples/tts/conf/hifigan/model/validation_ds/val_ds_finetune.yaml
@@ -12,3 +12,4 @@ dataloader_params:
   shuffle: false
   batch_size: 16
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/mixer-tts-x.yaml b/examples/tts/conf/mixer-tts-x.yaml
@@ -102,7 +102,7 @@ model:
       shuffle: true
       batch_size: 64
       num_workers: 4
-      pin_memory: false
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -131,7 +131,7 @@ model:
       shuffle: false
       batch_size: 64
       num_workers: 1
-      pin_memory: false
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor

diff --git a/examples/tts/conf/mixer-tts.yaml b/examples/tts/conf/mixer-tts.yaml
@@ -105,7 +105,7 @@ model:
       shuffle: true
       batch_size: 64
       num_workers: 4
-      pin_memory: false
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -133,7 +133,7 @@ model:
       shuffle: false
       batch_size: 64
       num_workers: 1
-      pin_memory: false
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor

diff --git a/examples/tts/conf/tacotron2.yaml b/examples/tts/conf/tacotron2.yaml
@@ -78,7 +78,7 @@ model:
       shuffle: true
       batch_size: 48
       num_workers: 4
-      pin_memory: false
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -105,7 +105,7 @@ model:
       shuffle: false
       batch_size: 24
       num_workers: 8
-      pin_memory: false
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.parts.preprocessing.features.FilterbankFeatures

diff --git a/examples/tts/conf/tacotron2_44100.yaml b/examples/tts/conf/tacotron2_44100.yaml
@@ -70,6 +70,7 @@ model:
       shuffle: true
       batch_size: 48
       num_workers: 4
+      pin_memory: true
 
 
   validation_ds:
@@ -90,6 +91,7 @@ model:
       shuffle: false
       batch_size: 48
       num_workers: 8
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.parts.preprocessing.features.FilterbankFeatures

diff --git a/examples/tts/conf/univnet/model/train_ds/train_ds.yaml b/examples/tts/conf/univnet/model/train_ds/train_ds.yaml
@@ -10,3 +10,4 @@ dataloader_params:
   shuffle: true
   batch_size: 32
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/univnet/model/train_ds/train_ds_finetune.yaml b/examples/tts/conf/univnet/model/train_ds/train_ds_finetune.yaml
@@ -12,3 +12,4 @@ dataloader_params:
   shuffle: true
   batch_size: 32
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/univnet/model/validation_ds/val_ds.yaml b/examples/tts/conf/univnet/model/validation_ds/val_ds.yaml
@@ -10,3 +10,4 @@ dataloader_params:
   shuffle: false
   batch_size: 16
   num_workers: 1
+  pin_memory: true
diff --git a/examples/tts/conf/univnet/model/validation_ds/val_ds_finetune.yaml b/examples/tts/conf/univnet/model/validation_ds/val_ds_finetune.yaml
@@ -12,3 +12,4 @@ dataloader_params:
   shuffle: false
   batch_size: 16
   num_workers: 4
+  pin_memory: true
diff --git a/examples/tts/conf/waveglow.yaml b/examples/tts/conf/waveglow.yaml
@@ -32,6 +32,7 @@ model:
       shuffle: true
       batch_size: 12
       num_workers: 4
+      pin_memory: true
 
   validation_ds:
     dataset:
@@ -45,6 +46,7 @@ model:
       shuffle: false
       batch_size: 8
       num_workers: 4
+      pin_memory: true
 
   preprocessor:
     _target_: nemo.collections.asr.parts.preprocessing.features.FilterbankFeatures