diff --git a/TTS/speaker_encoder/models/lstm.py b/TTS/speaker_encoder/models/lstm.py
index 7ac08514..ec394cdb 100644
--- a/TTS/speaker_encoder/models/lstm.py
+++ b/TTS/speaker_encoder/models/lstm.py
@@ -49,6 +49,7 @@ class LSTMSpeakerEncoder(nn.Module):
         self.use_lstm_with_projection = use_lstm_with_projection
         self.use_torch_spec = use_torch_spec
         self.audio_config = audio_config
+        self.proj_dim = proj_dim
 
         layers = []
         # choise LSTM layer
diff --git a/TTS/speaker_encoder/models/resnet.py b/TTS/speaker_encoder/models/resnet.py
index 643449c8..d6c3dad4 100644
--- a/TTS/speaker_encoder/models/resnet.py
+++ b/TTS/speaker_encoder/models/resnet.py
@@ -95,6 +95,7 @@ class ResNetSpeakerEncoder(nn.Module):
         self.log_input = log_input
         self.use_torch_spec = use_torch_spec
         self.audio_config = audio_config
+        self.proj_dim = proj_dim
 
         self.conv1 = nn.Conv2d(1, num_filters[0], kernel_size=3, stride=1, padding=1)
         self.relu = nn.ReLU(inplace=True)