-
-
Notifications
You must be signed in to change notification settings - Fork 66
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
Implement EfficientNet #171
Conversation
Probably using the same technique as #164. I think any model that is plain convolutions, batch norm, and fully connected can be done in this way. There a gist linked on the HF repo model cards if you want to tackle adding this for other models in the repo. |
Test errors are happening later for the same memory issues as always. I've already limited the tests to the smallest subset that seems necessary. |
Seems all good, but I'm worried that EfficientNets, which are the least heavy of the models, are causing this....or maybe it's a cumulative effect from the other models (Is |
The tests get passed EfficientNet and error around ConvNeXt. Maybe ConvNeXt is right on the threshold for the CI? |
Might be the |
Seems like it was the MLPMixers after all, nice! |
Could maybe try not testing for the |
Follow-up from #106 (closes #106).