Is there any modification made to the type of  embedding_dropout or Dropout in the Model. Are you reinstantiating somewhere? Also, can you display the assignment of embedding_dropout to self.embedding_dropout