fengyang0317 / unsupervised_captioning

Code for Unsupervised Image Captioning
MIT License
215 stars 51 forks source link

Hi, yang, May I ask you a question? #24

Open fresh382227905 opened 4 years ago

fresh382227905 commented 4 years ago

If I want to use tensorflow v2.0 to run the code? what should I do? There's no tf.contrib. I tried to use tf.compat.v1, but it doesn't work. It seems that It doesn't support it. I am not able to change the version of cuda. Can you give some suggestions. Looking forward to your reply. Thanks!!!!

fengyang0317 commented 4 years ago

tf.contrib has been moved to other packages such as tf_addons, tf_slim. I will provide a tf2 compatible training code in the future.

fresh382227905 commented 4 years ago

As soon as you provided the code, please tell me, thank you, Yang!!