├── .github └── workflows │ └── main.yml ├── .gitignore ├── README.md ├── config ├── dataset_config │ ├── afhq_cat.yaml │ ├── celeba.yaml │ └── celebahq.yaml ├── main_config.yaml └── method_config │ ├── d_flow.yaml │ ├── flow_priors.yaml │ ├── ot_ode.yaml │ ├── pnp_diff.yaml │ ├── pnp_flow.yaml │ └── pnp_gs.yaml ├── demo ├── algo.png ├── demo.ipynb ├── dirichlet │ ├── Diri_DFlow.ipynb │ └── Diri_PnP.ipynb ├── example.jpg └── pnp_flow.png ├── download.sh ├── main.py ├── pnpflow ├── __init__.py ├── compute_metric.py ├── dataloaders.py ├── degradations.py ├── fid_score.py ├── image_generation │ ├── __init__.py │ ├── configs │ │ ├── __init__.py │ │ ├── default_cifar10_configs.py │ │ ├── default_lsun_configs.py │ │ └── rectified_flow │ │ │ ├── afhq_cat_pytorch_rf_gaussian.py │ │ │ ├── bedroom_rf_gaussian.py │ │ │ ├── celeba_hq_pytorch_rf_gaussian.py │ │ │ ├── church_rf_gaussian.py │ │ │ ├── cifar10_rf_gaussian_ddpmpp.py │ │ │ ├── cifar10_rf_gaussian_reflow_distill_k=1.py │ │ │ ├── cifar10_rf_gaussian_reflow_distill_k=1_online.py │ │ │ ├── cifar10_rf_gaussian_reflow_distill_k_g_1.py │ │ │ ├── cifar10_rf_gaussian_reflow_generate_data.py │ │ │ ├── cifar10_rf_gaussian_reflow_train.py │ │ │ └── cifar10_rf_gaussian_reflow_train_online.py │ ├── datasets.py │ ├── debug.py │ ├── evaluation.py │ ├── likelihood.py │ ├── losses.py │ ├── main.py │ ├── models │ │ ├── __init__.py │ │ ├── ddpm.py │ │ ├── ema.py │ │ ├── layers.py │ │ ├── layerspp.py │ │ ├── ncsnpp.py │ │ ├── ncsnv2.py │ │ ├── normalization.py │ │ ├── up_or_down_sampling.py │ │ └── utils.py │ ├── op │ │ ├── __init__.py │ │ ├── fused_act.py │ │ ├── fused_bias_act.cpp │ │ ├── fused_bias_act_kernel.cu │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.py │ │ └── upfirdn2d_kernel.cu │ ├── pytorch_datasets.py │ ├── run_lib.py │ ├── run_lib_pytorch.py │ ├── run_lib_reflow.py │ ├── sampling.py │ ├── sde_lib.py │ └── utils.py ├── methods │ ├── __init__.py │ ├── d_flow.py │ ├── flow_priors.py │ ├── ot_ode.py │ ├── pnp_diff.py │ ├── pnp_flow.py │ └── pnp_gs.py ├── models.py ├── tests │ └── test_unit.py ├── toy_example.py ├── train_denoiser.py ├── train_flow_matching.py └── utils.py ├── pyproject.toml ├── requirements.txt ├── scripts ├── afhq_validation_images.sh ├── script_compute_time.sh ├── script_test.sh ├── script_train.sh └── script_val.sh └── splits └── afhq_cat └── validation_images.txt /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/README.md -------------------------------------------------------------------------------- /config/dataset_config/afhq_cat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/dataset_config/afhq_cat.yaml -------------------------------------------------------------------------------- /config/dataset_config/celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/dataset_config/celeba.yaml -------------------------------------------------------------------------------- /config/dataset_config/celebahq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/dataset_config/celebahq.yaml -------------------------------------------------------------------------------- /config/main_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/main_config.yaml -------------------------------------------------------------------------------- /config/method_config/d_flow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/d_flow.yaml -------------------------------------------------------------------------------- /config/method_config/flow_priors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/flow_priors.yaml -------------------------------------------------------------------------------- /config/method_config/ot_ode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/ot_ode.yaml -------------------------------------------------------------------------------- /config/method_config/pnp_diff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/pnp_diff.yaml -------------------------------------------------------------------------------- /config/method_config/pnp_flow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/pnp_flow.yaml -------------------------------------------------------------------------------- /config/method_config/pnp_gs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/config/method_config/pnp_gs.yaml -------------------------------------------------------------------------------- /demo/algo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/algo.png -------------------------------------------------------------------------------- /demo/demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/demo.ipynb -------------------------------------------------------------------------------- /demo/dirichlet/Diri_DFlow.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/dirichlet/Diri_DFlow.ipynb -------------------------------------------------------------------------------- /demo/dirichlet/Diri_PnP.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/dirichlet/Diri_PnP.ipynb -------------------------------------------------------------------------------- /demo/example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/example.jpg -------------------------------------------------------------------------------- /demo/pnp_flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/demo/pnp_flow.png -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/download.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/main.py -------------------------------------------------------------------------------- /pnpflow/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = 0.1 2 | -------------------------------------------------------------------------------- /pnpflow/compute_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/compute_metric.py -------------------------------------------------------------------------------- /pnpflow/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/dataloaders.py -------------------------------------------------------------------------------- /pnpflow/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/degradations.py -------------------------------------------------------------------------------- /pnpflow/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/fid_score.py -------------------------------------------------------------------------------- /pnpflow/image_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/default_cifar10_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/default_cifar10_configs.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/default_lsun_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/default_lsun_configs.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/afhq_cat_pytorch_rf_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/afhq_cat_pytorch_rf_gaussian.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/bedroom_rf_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/bedroom_rf_gaussian.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/celeba_hq_pytorch_rf_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/celeba_hq_pytorch_rf_gaussian.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/church_rf_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/church_rf_gaussian.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_ddpmpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_ddpmpp.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1_online.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1_online.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k_g_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k_g_1.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_generate_data.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train.py -------------------------------------------------------------------------------- /pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train_online.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train_online.py -------------------------------------------------------------------------------- /pnpflow/image_generation/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/datasets.py -------------------------------------------------------------------------------- /pnpflow/image_generation/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/debug.py -------------------------------------------------------------------------------- /pnpflow/image_generation/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/evaluation.py -------------------------------------------------------------------------------- /pnpflow/image_generation/likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/likelihood.py -------------------------------------------------------------------------------- /pnpflow/image_generation/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/losses.py -------------------------------------------------------------------------------- /pnpflow/image_generation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/main.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pnpflow/image_generation/models/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/ddpm.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/ema.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/layers.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/layerspp.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/ncsnpp.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/ncsnv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/ncsnv2.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/normalization.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/up_or_down_sampling.py -------------------------------------------------------------------------------- /pnpflow/image_generation/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/models/utils.py -------------------------------------------------------------------------------- /pnpflow/image_generation/op/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pnpflow/image_generation/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/fused_act.py -------------------------------------------------------------------------------- /pnpflow/image_generation/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /pnpflow/image_generation/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /pnpflow/image_generation/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /pnpflow/image_generation/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/upfirdn2d.py -------------------------------------------------------------------------------- /pnpflow/image_generation/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /pnpflow/image_generation/pytorch_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/pytorch_datasets.py -------------------------------------------------------------------------------- /pnpflow/image_generation/run_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/run_lib.py -------------------------------------------------------------------------------- /pnpflow/image_generation/run_lib_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/run_lib_pytorch.py -------------------------------------------------------------------------------- /pnpflow/image_generation/run_lib_reflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/run_lib_reflow.py -------------------------------------------------------------------------------- /pnpflow/image_generation/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/sampling.py -------------------------------------------------------------------------------- /pnpflow/image_generation/sde_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/sde_lib.py -------------------------------------------------------------------------------- /pnpflow/image_generation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/image_generation/utils.py -------------------------------------------------------------------------------- /pnpflow/methods/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pnpflow/methods/d_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/d_flow.py -------------------------------------------------------------------------------- /pnpflow/methods/flow_priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/flow_priors.py -------------------------------------------------------------------------------- /pnpflow/methods/ot_ode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/ot_ode.py -------------------------------------------------------------------------------- /pnpflow/methods/pnp_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/pnp_diff.py -------------------------------------------------------------------------------- /pnpflow/methods/pnp_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/pnp_flow.py -------------------------------------------------------------------------------- /pnpflow/methods/pnp_gs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/methods/pnp_gs.py -------------------------------------------------------------------------------- /pnpflow/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/models.py -------------------------------------------------------------------------------- /pnpflow/tests/test_unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/tests/test_unit.py -------------------------------------------------------------------------------- /pnpflow/toy_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/toy_example.py -------------------------------------------------------------------------------- /pnpflow/train_denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/train_denoiser.py -------------------------------------------------------------------------------- /pnpflow/train_flow_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/train_flow_matching.py -------------------------------------------------------------------------------- /pnpflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pnpflow/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/afhq_validation_images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/scripts/afhq_validation_images.sh -------------------------------------------------------------------------------- /scripts/script_compute_time.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/scripts/script_compute_time.sh -------------------------------------------------------------------------------- /scripts/script_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/scripts/script_test.sh -------------------------------------------------------------------------------- /scripts/script_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/scripts/script_train.sh -------------------------------------------------------------------------------- /scripts/script_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/scripts/script_val.sh -------------------------------------------------------------------------------- /splits/afhq_cat/validation_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/annegnx/PnP-Flow/HEAD/splits/afhq_cat/validation_images.txt --------------------------------------------------------------------------------