Skip to content

Custom classes added for phy_VAE#48

Open
arpanbiswas52 wants to merge 7 commits intoziatdinovmax:mainfrom
arpanbiswas52:main
Open

Custom classes added for phy_VAE#48
arpanbiswas52 wants to merge 7 commits intoziatdinovmax:mainfrom
arpanbiswas52:main

Conversation

@arpanbiswas52
Copy link
Copy Markdown

Added custom classes under trainer and model folders to extend the application of physics driven VAE. Please review

custom svi trainer class to incorporate physics driven loss functions
SVI class extended to custom svi class to incorporate custom loss with VAE loss during training process
traceELBO class extended to custom_traceELBO to incorporate physics driven loss with VAE loss
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.

1 participant