diff --git a/data_utils.py b/data_utils.py
index 7285672..bf52ba6 100644
--- a/data_utils.py
+++ b/data_utils.py
@@ -29,6 +29,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
         self.win_length = hparams.win_length
         self.sampling_rate = hparams.sampling_rate
         self.spk_map = hparams.spk2id
+        self.hparams = hparams
 
         self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False)
         if self.use_mel_spec_posterior: