Skip to content

Try VAE with AE reconstruction loss - #2

Merged
noahpy merged 14 commits into
masterfrom
vae-ae
Oct 10, 2025
Merged

Try VAE with AE reconstruction loss#2
noahpy merged 14 commits into
masterfrom
vae-ae

Conversation

@noahpy

@noahpy noahpy commented Oct 10, 2025

Copy link
Copy Markdown
Owner

Tried VAE for MNIST with additional AE reconstruciton loss constidered to total loss. No significant improvment was found.

Changes to Code

  • Tuning can now suppport Evaluator result for comparison metric
  • BaseModel:calc_loss() can now also return a dict containing compoenents of loss for better debugging

@noahpy
noahpy merged commit 86bf302 into master Oct 10, 2025
4 checks passed
@noahpy
noahpy deleted the vae-ae branch October 10, 2025 08:23
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