Change the Samplers weights to float for save memory

coqui-ai · Feb 11, 2022 · e8d710f · e8d710f
1 parent 525eb93
commit e8d710f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/TTS/tts/utils/languages.py b/TTS/tts/utils/languages.py
@@ -121,4 +121,4 @@ def get_language_balancer_weights(items: list):
     dataset_samples_weight = np.array([weight_language[l] for l in language_ids])
     # normalize
     dataset_samples_weight = dataset_samples_weight / np.linalg.norm(dataset_samples_weight)
-    return torch.from_numpy(dataset_samples_weight).double()
+    return torch.from_numpy(dataset_samples_weight).float()
diff --git a/TTS/tts/utils/speakers.py b/TTS/tts/utils/speakers.py
@@ -421,4 +421,4 @@ def get_speaker_balancer_weights(items: list):
     dataset_samples_weight = np.array([weight_speaker[l] for l in speaker_ids])
     # normalize
     dataset_samples_weight = dataset_samples_weight / np.linalg.norm(dataset_samples_weight)
-    return torch.from_numpy(dataset_samples_weight).double()
+    return torch.from_numpy(dataset_samples_weight).float()