Train and test image generation models