-
Notifications
You must be signed in to change notification settings - Fork 2.1k
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
Changes related to TF backend compatibility check for Vision Transformer on small datasets #1671
base: master
Are you sure you want to change the base?
Conversation
…ansformer on small datasets
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
Hi @sachinprasadhs, I am not sure why CLA check is failing. I and the team which handles corporate CLA matters at my employer, we all tried all steps mentioned. I was thinking you or anyone from google might be able to help me regarding this. |
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.
Thanks for the PR! The code looks good -- please also generate the md
and ipynb
files via the python autogen.py add_example vision/vit_small_ds
command (in scripts/
directory)
I am able to generate
When I updated to Tensorflow to 2.15.0 and Keras to 3.0.0 then it created We might need to update |
Can you add the md and ipynb files to the PR? |
Sure, I will be adding soon. I was able to generate but I noticed I had reduced dataset size and number of epochs for faster experimentation. I tried with full dataset and full epochs but kernal stopped after 24 hours in my macbook automatically at 46th epoch out of 50 epochs. So, I need to restart the generation process again. |
You're going to need a GPU VM (like a V100), running on CPU will take too much time. |
This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. |
I tried GPU version of tensorflow for mac M1 metal but still taking lot of time and getting crashed after 12 hours. I also rented GPU from jarvislabs.ai but they are not providing latest version of tensorflow in their image. It seems pretty hard for me to generate .md file due to not having access to Nvidia GPU. |
@fchollet Please review. |
@sachinprasadhs can you please review and merge these changes as this PR is opened from long time. |
@pksX01 , Can you generate .ipynb and .md files instead of making manual changes. |
Added generated .md and .ipynb files @sachinprasadhs can you please review? |
Can you exclude the workflow changes in this PR. |
Done. @sachinprasadhs |
This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. |
@sachinprasadhs can you please review it, it is marked stale. |
This PR caters to address changes required to convert vit_small_ds.py to Keras 3 with Tensorflow backend. This PR does not add backend_agnostic changes.