This repository has been archived by the owner on Jun 9, 2021. It is now read-only.
Certain kind of layers breaks: No MLCTrainingGraph has been found.
#23
Comments
Willian-Zhang
changed the title
Certain kinds of breaks during training:
Certain kind of layers breaks: Nov 20, 2020
No MLCTrainingGraph has been found.
No MLCTrainingGraph has been found.
@Willian-Zhang Thank you for reporting this failure and providing a reproducible test case. We will investigate. |
@Willian-Zhang, could you try the updated wheel and let us know if you still see the crash? Thank you! |
I can use tf.keras.layers.Dropout now, but I found a Weird bug: "dropout can not work on CPU mode". use CPU, can only get loss > 1000, and accuracy around 0.82 on test set and 0.90 on validation set after 3 epochs,
but same code use GPU can get loss less than 0.2 and accuracy around 0.95 on test set and 0.99 on validation set after 3 epochs.
and if we remove dropout layer, everything works fine. the code is here:
|
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
Adding
tf.keras.layers.Dropout
to model results the following error:Expected behavior:
Running
.fit
on model successExample model configuration:
The text was updated successfully, but these errors were encountered: