Closed mirfan899 closed 4 years ago
Hi,
I have just instantiated a model and loaded the pre-trained weights on Tensorflow 2.1 (tf-keras 2.2.4), the prediction works fine, I don't think it is a question of old weights (you could load and re-export them via tf-2.1 if you want to try).
Feels more like a cuda-related problem. Does the model works properly on CPU?
I've updated the Cudnn to 7.6.5 and Cuda 10 and TensorFlow 1.15 and it did not work. It seems I have to installed TensorFlow 2.1
Any update on your problem? If not I am closing the issue because it seems related to your configuration.
Yes, it resolved.
I've tried to run the example but it seems your weights are too old and cannot be used with the TensorFlow version(1.13.1), Keras 2.3.1.
Cuda 10.0.130 Cudnn 7.3.1