Flow Matching: Allow passing class as subnet (2.0 version) #200
Labels
feature
New feature or request
Comments
|
I like the idea and second the proposal. Would you mind drafting a PR that implements this? Based on your description it should be quite straightforward. |
vpratz
added a commit
to vpratz/BayesFlow
that referenced
this issue
Oct 9, 2024
* in addition to strings, type (i.e., uninstantiated classes) are accepted as well * allows easy use of user-defined networks
stefanradev93
pushed a commit
that referenced
this issue
Oct 9, 2024
|
closed by #201 |
stefanradev93
pushed a commit
that referenced
this issue
Oct 22, 2024
* Allow calling find_network with type (#200) * in addition to strings, type (i.e., uninstantiated classes) are accepted as well * allows easy use of user-defined networks * Allow tyoe in find_network: update type hints * Complete Consistency Model implementation * Add Consistency Model to networks * Consistency Models: minor fixes for Jax and TF * Consistency Models: Fix to the sampling routine * Documentation: Consistency Model example notebook * cosmetic updates to the consistency model notebook --------- Co-authored-by: Marvin Schmitt <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This is probably already planned:
FlowMatchingcurrently only accepts strings for thesubnetargument. Allowing classes as well would help with modularity, as it enables easily plugging in custom networks for power users, without sacrificing ease of use for others.Handling the case in
utils.find_networkand initializing the class there might be sufficient to enable this.The text was updated successfully, but these errors were encountered: