Skip to content

002 beta vae#3

Open
neogyk wants to merge 10 commits intobaler-collaboration:mainfrom
neogyk:002_beta_vae
Open

002 beta vae#3
neogyk wants to merge 10 commits intobaler-collaboration:mainfrom
neogyk:002_beta_vae

Conversation

@neogyk
Copy link
Copy Markdown

@neogyk neogyk commented Nov 19, 2024

Added the beta variational auto-encoder with linear layers

neogyk added 10 commits May 20, 2024 23:58
Updated the model definition.

Transformer AE class consume as the input:
Dimentions of the input, output and latent:
in_dim:int
out_dim:int
latent_dim:int

The Size of the hidden layers of the encoder and decoder parts:
encoder_h_dim:list
decoder_h_dim:list

nhead
activation fucntion
Updated the Variational AE. Added Ghost Batch Normalization Layer
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