UNet Flax modeling #485
Closed
UNet Flax modeling #485
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Mimic the structure of the PyTorch files. The model classes themselves need work, depending on what we do about configuration and initialization.
|
The documentation is not available anymore as the PR was closed or merged. |
This was referenced Sep 13, 2022
|
Superseded by #502. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
So far I've just put stuff in files mimicking the structure of the PyTorch codebase. The main topic that remains open is what to do with construction and initialization (do we need a wrapped
moduleto keep taking advantage of mixins in order to havefrom_pretrained?), as discussed in google/flax#2454.