search

taki0112 / UGATIT

Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)
MIT License
6.17k stars 1.04k forks source link

"tf.errors.InvalidArgumentError" Occured while running with "--ch 16 --img_size 64" #40

Closed FutaAlice closed 5 years ago

FutaAlice commented 5 years ago

Everything works well with default arguments, but training time seems too long on my GPU.

I attempted to use "--ch 16 --img_size 64 --" arguments and create a smaller model, but error occured while building model.

Any ideas about create a tiny model? Thanks a lot.

taki0112 commented 5 years ago

If the image size is too small, It will be error in discriminator. I recommend not to reduce.