Skip to content
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

merge resnet into master #111

Merged
merged 40 commits into from Nov 18, 2020
Merged

merge resnet into master #111

merged 40 commits into from Nov 18, 2020

Conversation

cgarciae
Copy link
Collaborator

@cgarciae cgarciae commented Nov 18, 2020

Merges #108 into master

cgarciae and others added 9 commits November 8, 2020 11:47
* added resnet18

* imagenet input pipeline, from https://github.com/google/flax

* experimental support for mixed precision

* full training script

* black + resnet test

* format black

* re-jit when loading a model for compability among platforms

* format black

* use different poetry installer

Co-authored-by: Cristian Garcia <cgarcia.e88@gmail.com>
@cgarciae cgarciae changed the title feature/custom_jit merge resnet into master Nov 18, 2020
@cgarciae cgarciae merged commit 9538853 into master Nov 18, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants