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

checkpoint InvalidArgumentError #48

Closed xizeyoupan closed 5 years ago

xizeyoupan commented 5 years ago

I've tried your checkpoint but an err occurred: tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [256,256] rhs shape= [1048576,256] [[{{node save/Assign_105}}]]

err, "a mismatch between the current graph and the graph")

tensorflow.python.framework.errors_impl.InvalidArgumentError: Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error: Assign requires shapes of both tensors to match. lhs shape= [256,256] rhs shape= [1048576,256]

what should I do? thx!

xizeyoupan commented 5 years ago

sorry,I mistake folder name