-
Notifications
You must be signed in to change notification settings - Fork 45.8k
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
NCF XLA and Eager tests with a refactor of resnet flags to make this cleaner. #7067
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
thank you for the cleanup!
Test failures are Transformer. I fixed all ResNet unit tests failures. Hopefully I or we can clean up the transformer issues soon as well. |
First: I am sorry this got big. What happened is to move run_eagerly as a based flag Resnet flag usage needed refactored. And I wanted to do it anyway to reduce our calls to resnet_run_loop from the TF 2.0/Keras code.
I have run ResNet50 and ResNet56 1 GPU tests benchmark.*1_gpu and for ResNet50 benchmark.*8_gpu and then the NCF tests.
Very soon I will factor out the rest of the calls to imagenet_main.py from the v2 folder so we can more easily move the v1 code into some archive folder. I think only the constants are left.