Skip to content

v0.2.1: Pipeline Training Mode & GAN-Only Training

Choose a tag to compare

@danny-mio danny-mio released this 09 Dec 20:06
· 69 commits to main since this release
Immutable release. Only release title and notes can be modified.
8fb8c49

πŸš€ Major Features

Pipeline Training Mode (NEW)

Multi-step sequential training with per-step configuration:

  • Sequential stages: Define warmup β†’ GAN β†’ flow in single YAML config
  • Per-step freeze/unfreeze: Selective component training
  • Loss-threshold transitions: Automatically progress when loss reaches target
  • Inline optimizer/scheduler configs: Different optimizers per step
  • Per-step checkpoints/metrics: Step-specific files for focused analysis
  • Full resume support: Mid-pipeline resume with state preservation

GAN-Only Training Mode (NEW)

  • train_reconstruction parameter (default: true)
  • Set to false for adversarial-only training
  • Use case: SPADE conditioning without reconstruction overhead

Enhanced Logging & Monitoring

  • Batch timing (Xs/batch)
  • Step-specific progress files
  • Correct GAN loss keys (loss_gen, loss_disc)
  • Mid-epoch sample generation with batch numbers

πŸ› Bug Fixes (12 commits)

  • Fixed encoder gradients for GAN-only mode
  • Fixed VAE trainer calling for GAN-only
  • Fixed EMA creation for GAN-only mode
  • Fixed checkpoint resume state tracking
  • Fixed diagram generation for pipeline metrics
  • Fixed sample file renaming

πŸ“š Documentation

  • docs/PIPELINE_ARCHITECTURE.md (547 lines)
  • Pipeline training in docs/TRAINING_GUIDE.md (100+ lines)
  • Pipeline testing in CONTRIBUTING.md
  • Complete CHANGELOG.md entry

πŸ§ͺ Testing

  • 446/446 tests passing βœ…
  • All linters clean βœ…

πŸ“¦ Installation

pip install fluxflow-training==0.2.1

πŸ”— Dependencies

Requires: fluxflow-core v0.2.1+

Full Changelog: v0.1.1...v0.2.1