MilanaVayantrub v2.0

Used epoch 1 in first version. There was some banding in the generated images. Retrained and am using epoch 4 in this version with no visible banding anymore.