Share my energy based model tutorial #1348
Unanswered
odie2630463
asked this question in
Show and tell
Replies: 1 comment 1 reply
-
|
Cool! I don't know much about EBMs but took quick glace at your Colab -- why not make swish simply a function? No need to make it a module as it has no variables or uses no RNGs. |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Hi, EBM is the energy-based model for fitting data distribution P(X). Swish roll is an unknown distribution and we only have the data samples from it. So we training the model on the data and synths, which is sampled by MCMC sampler from the model. Flax version is faster than Pytorch 50x. Awesome !!! |
Beta Was this translation helpful? Give feedback.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi Flax, I implement EBM in Flax, share with everyone, and if has any advice I would like to know. Thanks!
Beta Was this translation helpful? Give feedback.
All reactions