Skip to content

Add ResNet-based encoder, Perceptual loss, KLAnnealing, and ParamManager#260

Merged
BerndDoser merged 88 commits into
HITS-AIN:mainfrom
BerndDoser:gmr
May 12, 2026
Merged

Add ResNet-based encoder, Perceptual loss, KLAnnealing, and ParamManager#260
BerndDoser merged 88 commits into
HITS-AIN:mainfrom
BerndDoser:gmr

Conversation

@BerndDoser

Copy link
Copy Markdown
Member
  • Added a ResNet-based encoder, an upsampling decoder
  • Added a Perceptual loss
  • Added a new KLAnnealing callback (src/spherinator/callbacks/kl_annealing.py) for flexible annealing of the KL divergence weight during VAE training
  • Added a ParamManager callback (src/spherinator/callbacks/param_manager.py) for parameter value management and freezing based on regex patterns
  • Improved LogReconstructionCallback to handle datasets with augmentations, logging original, augmented, and reconstructed images in a single figure
  • Updated .github/workflows/python-package.yml to use the latest versions of GitHub Actions and setup-uv, improving CI reliability and compatibility
  • Added ruff checks for linting and formatting to .github/check.sh for improved code quality enforcement

…oad_checkpoint after the state dict is restored
BerndDoser and others added 27 commits April 21, 2026 11:09
Co-authored-by: Copilot <copilot@github.com>
@BerndDoser
BerndDoser merged commit 3a29e05 into HITS-AIN:main May 12, 2026
2 of 3 checks passed
@BerndDoser
BerndDoser deleted the gmr branch May 12, 2026 21:13
@BerndDoser BerndDoser added this to the v0.6 milestone May 29, 2026
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