-
Notifications
You must be signed in to change notification settings - Fork 198
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
RuntimeError when running baselines/imagenet/sngp.py #329
Comments
Unfortunately, I was not able to resolve this. I saw in GitHub issues in other repos that people who got the same error message were able to resolve it by specifying But I am not sure whether the on-read-synchronization is the correct behavior in this case.
|
Hey! Sorry for the delay. What was the command you were using to run the code, and what was the GPU/version of CUDA/version of TF? Also, have you been able to reproduce this at the current HEAD? |
I get the exact same error. Fresh install (nightly version of tf - 9/7/21), CUDA 11.2. |
I also get this error with TF 2.7 |
Dear uncertainty-baseline authors,
I am trying to run the SNGP training on ImageNet using uncertainty-baselines/baselines/imagenet/sngp.py.
It errors during the execution of the first training step with the following message:
This is the stack trace:
It seems like the
self.precision_matrix.assign
call inedward2/edward2/tensorflow/layers/random_feature.py
causes this error, because it is executed inside the strategy.run call of a tf.function.What can I do to fix this?
The text was updated successfully, but these errors were encountered: