Try VAE with additional AE reconstruction loss - #1

Merged

noahpy merged 13 commits into

Oct 10, 2025

Commits on Aug 22, 2025

calc_loss: allow return of dict of loss components
noahpy
committed
d2db11b
Introduce VAE with AE feedback
noahpy
committed
d76db7b
calc_loss: correct typing signature
noahpy
committed
c57f940
Update README.md
noahpy
authored
a41298e
Tuner: allow passing of Evaluator + save ranking models
noahpy
committed
70ad34f
VAE: softmax the loss weighing
noahpy
committed
52d6a51
Adapt on_eval() signatures
noahpy
committed
27ed62b
example/classifier/tune: add model path to dynamic tuning
noahpy
committed
8bac2ea
Introduce script for Tuning with VAE and Evaluator
noahpy
committed
2741e8a
Introduce VAE checking with AE reconstruction loss
noahpy
committed
32e1e80
hparams adjustments
noahpy
committed
a4e6428
save models
noahpy
committed
81e5e37
VAE: update image generation png
noahpy
committed
903e523