├── .gitignore ├── DSBM-Gaussian.py ├── README.md ├── bridge.def ├── bridge ├── __init__.py ├── data │ ├── __init__.py │ ├── afhq.py │ ├── cacheloader.py │ ├── downscaler.py │ ├── emnist.py │ ├── metrics.py │ └── utils.py ├── models │ ├── __init__.py │ ├── basic │ │ ├── __init__.py │ │ ├── basic_cond.py │ │ ├── layers.py │ │ └── time_embedding.py │ ├── ddpmpp │ │ ├── layers.py │ │ ├── layerspp.py │ │ ├── ncsnpp.py │ │ ├── normalization.py │ │ ├── op │ │ │ ├── __init__.py │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.py │ │ │ └── upfirdn2d_kernel.cu │ │ ├── up_or_down_sampling.py │ │ └── utils.py │ └── unet │ │ ├── __init__.py │ │ ├── fp16_util.py │ │ ├── layers.py │ │ └── unet.py ├── runners │ ├── __init__.py │ ├── config_getters.py │ ├── ema.py │ ├── logger.py │ ├── plotters.py │ └── repeater.py ├── sde │ ├── __init__.py │ ├── diffusion_bridge.py │ ├── discrete_langevin.py │ └── optimal_transport.py ├── trainer_dbdsb.py ├── trainer_dsb.py └── trainer_rf.py ├── conf ├── config.yaml ├── dataset │ ├── afhq_transfer.yaml │ ├── cifar10.yaml │ ├── downscaler_transfer.yaml │ └── mnist_transfer.yaml ├── gaussian.yaml ├── job.yaml ├── launcher │ ├── slurm_cpu.yaml │ └── slurm_gpu.yaml ├── method │ ├── bm.yaml │ ├── dbdsb.yaml │ ├── dbdsb_vp.yaml │ ├── dsb.yaml │ ├── otcfm.yaml │ └── rf.yaml ├── model │ ├── DDPMpp_32.yaml │ ├── DDPMpp_RF.yaml │ ├── DownscalerUNET.yaml │ └── UNET.yaml ├── test_config.yaml └── test_job.yaml ├── main.py ├── run_dbdsb.py ├── run_dsb.py ├── run_rf.py ├── test.py ├── test_dbdsb.py └── test_rf.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /DSBM-Gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/DSBM-Gaussian.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /bridge.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge.def -------------------------------------------------------------------------------- /bridge/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bridge/data/__init__.py: -------------------------------------------------------------------------------- 1 | from .cacheloader import * -------------------------------------------------------------------------------- /bridge/data/afhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/afhq.py -------------------------------------------------------------------------------- /bridge/data/cacheloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/cacheloader.py -------------------------------------------------------------------------------- /bridge/data/downscaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/downscaler.py -------------------------------------------------------------------------------- /bridge/data/emnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/emnist.py -------------------------------------------------------------------------------- /bridge/data/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/metrics.py -------------------------------------------------------------------------------- /bridge/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/data/utils.py -------------------------------------------------------------------------------- /bridge/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/__init__.py -------------------------------------------------------------------------------- /bridge/models/basic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bridge/models/basic/basic_cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/basic/basic_cond.py -------------------------------------------------------------------------------- /bridge/models/basic/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/basic/layers.py -------------------------------------------------------------------------------- /bridge/models/basic/time_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/basic/time_embedding.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/layers.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/layerspp.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/ncsnpp.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/normalization.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/op/__init__.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /bridge/models/ddpmpp/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/op/upfirdn2d.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /bridge/models/ddpmpp/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/up_or_down_sampling.py -------------------------------------------------------------------------------- /bridge/models/ddpmpp/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/ddpmpp/utils.py -------------------------------------------------------------------------------- /bridge/models/unet/__init__.py: -------------------------------------------------------------------------------- 1 | from .unet import * 2 | 3 | -------------------------------------------------------------------------------- /bridge/models/unet/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/unet/fp16_util.py -------------------------------------------------------------------------------- /bridge/models/unet/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/unet/layers.py -------------------------------------------------------------------------------- /bridge/models/unet/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/models/unet/unet.py -------------------------------------------------------------------------------- /bridge/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/__init__.py -------------------------------------------------------------------------------- /bridge/runners/config_getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/config_getters.py -------------------------------------------------------------------------------- /bridge/runners/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/ema.py -------------------------------------------------------------------------------- /bridge/runners/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/logger.py -------------------------------------------------------------------------------- /bridge/runners/plotters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/plotters.py -------------------------------------------------------------------------------- /bridge/runners/repeater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/runners/repeater.py -------------------------------------------------------------------------------- /bridge/sde/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/sde/__init__.py -------------------------------------------------------------------------------- /bridge/sde/diffusion_bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/sde/diffusion_bridge.py -------------------------------------------------------------------------------- /bridge/sde/discrete_langevin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/sde/discrete_langevin.py -------------------------------------------------------------------------------- /bridge/sde/optimal_transport.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/sde/optimal_transport.py -------------------------------------------------------------------------------- /bridge/trainer_dbdsb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/trainer_dbdsb.py -------------------------------------------------------------------------------- /bridge/trainer_dsb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/trainer_dsb.py -------------------------------------------------------------------------------- /bridge/trainer_rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/bridge/trainer_rf.py -------------------------------------------------------------------------------- /conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/config.yaml -------------------------------------------------------------------------------- /conf/dataset/afhq_transfer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/dataset/afhq_transfer.yaml -------------------------------------------------------------------------------- /conf/dataset/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/dataset/cifar10.yaml -------------------------------------------------------------------------------- /conf/dataset/downscaler_transfer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/dataset/downscaler_transfer.yaml -------------------------------------------------------------------------------- /conf/dataset/mnist_transfer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/dataset/mnist_transfer.yaml -------------------------------------------------------------------------------- /conf/gaussian.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/gaussian.yaml -------------------------------------------------------------------------------- /conf/job.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/job.yaml -------------------------------------------------------------------------------- /conf/launcher/slurm_cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/launcher/slurm_cpu.yaml -------------------------------------------------------------------------------- /conf/launcher/slurm_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/launcher/slurm_gpu.yaml -------------------------------------------------------------------------------- /conf/method/bm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/bm.yaml -------------------------------------------------------------------------------- /conf/method/dbdsb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/dbdsb.yaml -------------------------------------------------------------------------------- /conf/method/dbdsb_vp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/dbdsb_vp.yaml -------------------------------------------------------------------------------- /conf/method/dsb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/dsb.yaml -------------------------------------------------------------------------------- /conf/method/otcfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/otcfm.yaml -------------------------------------------------------------------------------- /conf/method/rf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/method/rf.yaml -------------------------------------------------------------------------------- /conf/model/DDPMpp_32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/model/DDPMpp_32.yaml -------------------------------------------------------------------------------- /conf/model/DDPMpp_RF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/model/DDPMpp_RF.yaml -------------------------------------------------------------------------------- /conf/model/DownscalerUNET.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/model/DownscalerUNET.yaml -------------------------------------------------------------------------------- /conf/model/UNET.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/model/UNET.yaml -------------------------------------------------------------------------------- /conf/test_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/test_config.yaml -------------------------------------------------------------------------------- /conf/test_job.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/conf/test_job.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/main.py -------------------------------------------------------------------------------- /run_dbdsb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/run_dbdsb.py -------------------------------------------------------------------------------- /run_dsb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/run_dsb.py -------------------------------------------------------------------------------- /run_rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/run_rf.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/test.py -------------------------------------------------------------------------------- /test_dbdsb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/test_dbdsb.py -------------------------------------------------------------------------------- /test_rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuyang-shi/dsbm-pytorch/HEAD/test_rf.py --------------------------------------------------------------------------------