search

google-research / xmcgan_image_generation

98 stars 15 forks source link

FailedPreconditionError #15

Open hyeonjinXZ opened 2 years ago

hyeonjinXZ commented 2 years ago

how to solve 'tensorflow.python.framework.errors_impl.FailedPreconditionError: StatelessRandomGetKeyCounter is stateful. [Op:SerializeIterator]'?

image

Below is my checkpoint directory: image

zzy994491827 commented 2 years ago

Can I ask your jax and jaxlib version? I encountered some bugs that were difficult to solve when training the model.

hyeonjinXZ commented 2 years ago

I used below version:

flax 0.3.6 jax 0.2.25 jaxlib 0.1.74+cuda11.cudnn82