Keep proj_dim in speaker encoder models

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

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)