.env.example
.gitignore
CHANGES.md
CLAUDE.md
CONFIG_USAGE.md
LICENSE
QUICKSTART_VAE.md
README.md
TODO_upload_hf.md
TRAINING_GUIDE.md
TRAIN_VAE_NOW.md
VAE_MIMIC_TRAINING.md
VAE_SCRIPTS_COMPARISON.md
VAE_SUMMARY.md
generate_example.py
pyproject.toml
upload_checkpoints_example.py
.github/workflows/publish.yml
configs/upload_checkpoints.yaml
configs/dataset/ptbxl.yaml
configs/experiments/pulse2pulse_mimic.yaml
configs/experiments/vae_mimic.yaml
configs/model/imn.yaml
configs/trainer/default.yaml
docs/VAE_MODEL.md
docs/VAE_QUICKSTART.md
runs/pulse2pulse_mimic/seed_42/config_resolved.yaml
runs/pulse2pulse_mimic/seed_42/metadata.json
runs/pulse2pulse_mimic/seed_42/tb/version_0/events.out.tfevents.1772121525.mimir.3649231.0
runs/pulse2pulse_mimic/seed_42/tb/version_0/hparams.yaml
scripts/README_VAE_MIMIC.md
scripts/README_VAE_TRAINING.md
scripts/export_predictions.sh
scripts/generate_pulse2pulse.py
scripts/run_sweep.sh
scripts/run_train_pulse2pulse_mimic.sh
scripts/run_train_pulse2pulse_standalone.sh
scripts/run_train_vae_mimic_config.sh
scripts/test_models_only.py
scripts/test_pulse2pulse_setup.py
scripts/train_pulse2pulse.py
scripts/train_vae_mimic.py
scripts/upload_checkpoints_to_hf.py
src/ecgen/__init__.py
src/ecgen/_version.py
src/ecgen/generate.py
src/ecgen/plot.py
src/ecgen.egg-info/PKG-INFO
src/ecgen.egg-info/SOURCES.txt
src/ecgen.egg-info/dependency_links.txt
src/ecgen.egg-info/requires.txt
src/ecgen.egg-info/top_level.txt
src/ecgen/data/datamodule.py
src/ecgen/data/mimic_dataset.py
src/ecgen/data/pulse2pulse_mimic.py
src/ecgen/data/transforms.py
src/ecgen/f/__init__.py
src/ecgen/f/upload_hf.py
src/ecgen/models/__init__.py
src/ecgen/models/pulse2pulse.py
src/ecgen/models/vae.py
src/ecgen/training/callbacks.py
src/ecgen/training/losses.py
src/ecgen/training/metrics.py
src/ecgen/training/test.py
src/ecgen/training/train.py
src/ecgen/training/validate.py
src/ecgen/utils/io.py
src/ecgen/utils/logging.py
src/ecgen/utils/metadata.py
src/ecgen/utils/seed.py
tests/test_data.py
tests/test_models.py
tests/test_upload_hf.py
tests/test_vae.py