Hi thanks for sharing your work, I have a small dataset including images and descriptions, Can I use this code for training on my dataset?