Skip to content

Implementation of the 'Knowing When to Look: Adaptive Attention via A Visual Sentinel for Image Captioning' by Lu et. al.

Notifications You must be signed in to change notification settings

xyy19920105/Adaptive-Sentinel-Image-Captioning

 
 

Repository files navigation

Adaptive-Sentinel-Image-Captioning

PyTorch implementation of the Knowing When to Look: Adaptive Attention via A Visual Sentinel for Image Captioning by Lu et. al.

Work in Progress.

TO-DO :

  • Implement normal LSTM-Attention
  • Greedy + Sampling Decoder
  • Layer Normalization
  • [] Debug Sentinel-LSTM
  • [] Implement Beam Search
  • [] Tune hyperparameters

Train Loss curve without pre-trained embeddings

alt text

Train Loss curve with pretrained embeddings

alt text

About

Implementation of the 'Knowing When to Look: Adaptive Attention via A Visual Sentinel for Image Captioning' by Lu et. al.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%