Pytorch Image Captioning Download the Flickr8k dataset Extract to folder dataset trainer.py for training the network. tester.py for testing the network. src/architecture.py for creating encoder, decoder and hooking them together. Reference