question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

compatibility issue (training error/accuracy) with tensorflow 1.5

See original GitHub issue

Everything works fine with tensorflow 1.4.

For newly released tensorflow 1.5, the training loss became NaN at 80 iterations:

iter: 80 / 84000, total loss: nan
 >>> rpn_loss_cls: 0.684417
 >>> rpn_loss_box: 0.019884
 >>> loss_cls: 0.101328
 >>> loss_box: 0.000000
 >>> lr: 0.001000

Issue Analytics

  • State:closed
  • Created 6 years ago
  • Comments:5

github_iconTop GitHub Comments

1reaction
dnnsparkcommented, Jan 31, 2018

@jwnsu I’m using AdamOptimizer. The NaN problem goes away when I use cpu version of tensorflow 1.5. This makes me think it may be a bug of new cuda9 / cudnn7.

1reaction
jwnsucommented, Jan 31, 2018

It seems tensorflow 1.5 affects training in subtle way. Regarding the error, it goes away after minor adjustment (e.g. learning rate.)

Read more comments on GitHub >

github_iconTop Results From Across the Web

TensorFlow version compatibility
This document is for users who need backwards compatibility across different versions of TensorFlow (either for code or data), ...
Read more >
TensorFlow: Dramatic loss of accuracy after freezing graph?
Potential source of problem: I first trained my model using TF 0.12, but I believe it is compatible with Tf 1.01, the version...
Read more >
tensorflow-metal | Apple Developer Forums
But still facing the GPU problem when training a 3D Unet. Here's part of my code and hoping to receive some suggestion to...
Read more >
Train With Mixed Precision - NVIDIA Documentation Center
The network accuracy was achieved from training in FP32. ... list of supported drivers, see the CUDA Application Compatibility topic.
Read more >
How to correctly install Keras and Tensorflow - ActiveState
Click to install Keras and Tensorflow together using pip. Understand how to use these Python libraries for machine learning use cases.
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found