Image Captioning System in Pytorch