.pre-commit-config.yaml
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
deployment/__init__.py
deployment/opensr_hpc/__init__.py
deployment/opensr_hpc/aoi.py
deployment/opensr_hpc/checkpoint.py
deployment/opensr_hpc/cli.py
deployment/opensr_hpc/collect.py
deployment/opensr_hpc/config.py
deployment/opensr_hpc/inference.py
deployment/opensr_hpc/logging_utils.py
deployment/opensr_hpc/manifests.py
deployment/opensr_hpc/metadata.py
deployment/opensr_hpc/naming.py
deployment/opensr_hpc/patching.py
deployment/opensr_hpc/raster.py
deployment/opensr_hpc/run_task.py
deployment/opensr_hpc/slurm.py
deployment/opensr_hpc/staging.py
deployment/opensr_hpc/submit.py
deployment/opensr_hpc/slurm/slurm_task_entrypoint.sh
docs/changelog.md
docs/training.md
opensr_model/__init__.py
opensr_model/spectral_correction.py
opensr_model/srmodel.py
opensr_model/utils.py
opensr_model.egg-info/PKG-INFO
opensr_model.egg-info/SOURCES.txt
opensr_model.egg-info/dependency_links.txt
opensr_model.egg-info/entry_points.txt
opensr_model.egg-info/requires.txt
opensr_model.egg-info/top_level.txt
opensr_model/ablations/__init__.py
opensr_model/ablations/temperature_eta.py
opensr_model/autoencoder/__init__.py
opensr_model/autoencoder/autoencoder.py
opensr_model/autoencoder/utils.py
opensr_model/configs/config_10m.yaml
opensr_model/configs/training_configs/raw_s2_examples.json
opensr_model/configs/training_configs/sen2neon_clean192.json
opensr_model/configs/training_configs/train_autoencoder.yaml
opensr_model/configs/training_configs/train_diffusion.yaml
opensr_model/configs/training_configs/train_sen2naip.yaml
opensr_model/denoiser/__init__.py
opensr_model/denoiser/unet.py
opensr_model/denoiser/utils.py
opensr_model/diffusion/__init__.py
opensr_model/diffusion/dpm_solver.py
opensr_model/diffusion/latentdiffusion.py
opensr_model/diffusion/utils.py
opensr_model/training/__init__.py
opensr_model/training/cli.py
opensr_model/training/data/__init__.py
opensr_model/training/data/mixture.py
opensr_model/training/data/taco.py
opensr_model/training/modules/__init__.py
opensr_model/training/modules/autoencoder.py
opensr_model/training/modules/diffusion.py
opensr_model/training/modules/losses.py
opensr_model/training/modules/metrics.py
opensr_model/training/modules/optim.py
opensr_model/training/runtime/__init__.py
opensr_model/training/runtime/callbacks.py
opensr_model/training/runtime/campaign_logging.py
opensr_model/training/runtime/checkpoints.py
opensr_model/training/runtime/config.py
opensr_model/training/runtime/provenance.py
opensr_model/training/validation/__init__.py
opensr_model/training/validation/callback.py
opensr_model/training/validation/data.py
opensr_model/training/validation/images.py
opensr_model/training/validation/metrics.py
resources/images/example.png
resources/images/example2.png
resources/images/example3.png
resources/images/gpu_instructions.png
resources/images/ldsr-s2_schema.png
resources/images/model_repo_icon.png
resources/images/opensr_logo.png
resources/images/run_all_tip.png
resources/images/sr_example.png
resources/images/uncertainty_map.png
resources/notebooks/(public)_LDSR_S2_No_Code.ipynb
resources/notebooks/(public)_LDSR_S2_Walkthrough.ipynb
resources/notebooks/(public)_LDSR_S2_sampling.ipynb
resources/notebooks/(public)_OpenSR_Showcase.ipynb
resources/sampler_info/sampler_measurements.csv
resources/sampler_info/sampler_speed_quality.png
resources/sampler_info/v1_sampler_comparison.png
tests/conftest.py
tests/ablations/test_temperature_eta.py
tests/autoencoder/test_autoencoder_core.py
tests/denoiser/test_components.py
tests/diffusion/test_diffusion_core.py
tests/diffusion/test_dpm_solver.py
tests/diffusion/test_latent_contracts.py
tests/diffusion/test_noise_schedules.py
tests/inference/test_checkpoint_io.py
tests/inference/test_inference.py
tests/inference/test_inference_utils.py
tests/spectral/test_spectral_correction.py
tests/spectral/test_spectral_inference.py
tests/training/conftest.py
tests/training/test_auxiliary_objectives.py
tests/training/test_callback_contracts.py
tests/training/test_checkpoint_contracts.py
tests/training/test_checkpoints.py
tests/training/test_cli.py
tests/training/test_config_cli.py
tests/training/test_cost_logging.py
tests/training/test_data.py
tests/training/test_data_integrity.py
tests/training/test_data_units.py
tests/training/test_diffusion_sharpness.py
tests/training/test_invalid_inputs.py
tests/training/test_srmodel_metadata.py
tests/training/test_training_layout.py
tests/training/test_training_regressions.py
tests/training/test_training_smoke.py
tests/training/test_training_units.py
tests/training/test_transfer_distributed.py
tests/training/test_transfer_validation.py
tests/training/test_validation_contracts.py
tests/training/test_validation_lifecycle.py