pritish commited on
Commit
3dd12be
1 Parent(s): 5cf2822

added one more weights

Browse files
model.py CHANGED
@@ -321,8 +321,8 @@ def get_caption_model():
321
  caption_model.decoder(sample_y, sample_enc_out, training=False)
322
 
323
  try:
324
- caption_model.load_weights('saved_models/image_captioning_transformer_weights_1.h5')
325
  except FileNotFoundError:
326
- caption_model.load_weights('Image-Captioning/saved_models/image_captioning_transformer_weights_1.h5')
327
 
328
  return caption_model
 
321
  caption_model.decoder(sample_y, sample_enc_out, training=False)
322
 
323
  try:
324
+ caption_model.load_weights('saved_models/image_captioning_transformer_weights_3.h5')
325
  except FileNotFoundError:
326
+ caption_model.load_weights('Image-Captioning/saved_models/image_captioning_transformer_weights_3.h5')
327
 
328
  return caption_model
saved_models/image_captioning_transformer_weights_3.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61d25373a21ccddf8bf24f67b2a1bb2037d565f8e88c85e20e7f5ef8527a90d9
3
+ size 247484996