Fix README, docs captioning example

This commit is contained in:
Francois Chollet 2015-11-22 12:05:26 -08:00
parent 252db48746
commit 05aecbc0bc
2 changed files with 2 additions and 2 deletions

@ -165,7 +165,7 @@ image_model.load_weights('weight_file.h5')
language_model = Sequential()
language_model.add(Embedding(vocab_size, 256, input_length=max_caption_len))
language_model.add(GRU(output_dim=128, return_sequences=True))
language_model.add(Dense(128))
language_model.add(TimeDistributedDense(128))
# let's repeat the image vector to turn it into a sequence.
image_model.add(RepeatVector(max_caption_len))

@ -140,7 +140,7 @@ image_model.load_weights('weight_file.h5')
language_model = Sequential()
language_model.add(Embedding(vocab_size, 256, input_length=max_caption_len))
language_model.add(GRU(output_dim=128, return_sequences=True))
language_model.add(Dense(128))
language_model.add(TimeDistributedDense(128))
# let's repeat the image vector to turn it into a sequence.
image_model.add(RepeatVector(max_caption_len))