Add files using upload-large-folder tool

README.md

@@ -0,0 +1,22 @@
+# CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200
+
+This repository contains model artifacts and configuration files from the CFM_CORR_EMA_50k experiment.
+
+## Contents
+
+This folder contains:
+- Model checkpoints and weights
+- Configuration files (JSON)
+- Scheduler and UNet components
+- Training results and metadata
+- Sample directories (excluding image files)
+
+## Experiment Details
+
+- Experiment: CFM_CORR_EMA_50k
+- Folder: CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200
+- Images excluded from upload
+
+## Usage
+
+This repository contains the necessary files to reproduce or analyze the model from this specific experimental run.

cfm_corr_400_ss0.01_ep80_ema-run0_998-800-600-400-200/config.json

@@ -0,0 +1,46 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "config.json",
+    "optim_name": "adamw",
+    "shed_name": "poly_decay",
+    "sched_num_warmup_steps": 45000,
+    "optim_weight_decay": 0.0,
+    "optim_beta_1": 0.9,
+    "optim_beta_2": 0.999,
+    "optim_epsilon": 1e-08,
+    "sched_lr_end": 1e-07,
+    "sched_power": 1.0,
+    "sched_t_max": 200,
+    "sched_last_epoch": -1,
+    "training_result_file": "training_result.json",
+    "check_finished_file": "finished.json",
+    "project": "CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200",
+    "name_postfix": "ema-run0_998-800-600-400-200",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "FrankCCCCC/ddpm_ema_cifar10",
+    "train_dataset": "cifar10",
+    "traj_dataset": "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/traj_dataset/TRAJ-DS_CORR-STEP-998-800-600_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000",
+    "batch_size": 256,
+    "num_epochs": 80,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 100,
+    "sigma_min": 0.0,
+    "sigma_src": 0.01,
+    "corr_trained_timestep": 400,
+    "ds_size": 50000,
+    "seed": 42,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

cfm_corr_400_ss0.01_ep80_ema-run0_998-800-600-400-200/finished.json

@@ -0,0 +1,3 @@
+{
+    "Msg": "Training finished."
+}

cfm_corr_400_ss0.01_ep80_ema-run0_998-800-600-400-200/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    400
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_400_ss0.01_ep80_ema-run0_998-800-600-400-200/training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 65,
+    "best_step": 12675
+}

cfm_corr_600_ss0.01_ep80_ema-run0_998-800-600-400-200/config.json

@@ -0,0 +1,46 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "config.json",
+    "optim_name": "adamw",
+    "shed_name": "poly_decay",
+    "sched_num_warmup_steps": 45000,
+    "optim_weight_decay": 0.0,
+    "optim_beta_1": 0.9,
+    "optim_beta_2": 0.999,
+    "optim_epsilon": 1e-08,
+    "sched_lr_end": 1e-07,
+    "sched_power": 1.0,
+    "sched_t_max": 200,
+    "sched_last_epoch": -1,
+    "training_result_file": "training_result.json",
+    "check_finished_file": "finished.json",
+    "project": "CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200",
+    "name_postfix": "ema-run0_998-800-600-400-200",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "FrankCCCCC/ddpm_ema_cifar10",
+    "train_dataset": "cifar10",
+    "traj_dataset": "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/traj_dataset/TRAJ-DS_CORR-STEP-998-800_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000",
+    "batch_size": 256,
+    "num_epochs": 80,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 100,
+    "sigma_min": 0.0,
+    "sigma_src": 0.01,
+    "corr_trained_timestep": 600,
+    "ds_size": 50000,
+    "seed": 42,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

cfm_corr_600_ss0.01_ep80_ema-run0_998-800-600-400-200/finished.json

@@ -0,0 +1,3 @@
+{
+    "Msg": "Training finished."
+}

cfm_corr_600_ss0.01_ep80_ema-run0_998-800-600-400-200/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    600
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_600_ss0.01_ep80_ema-run0_998-800-600-400-200/training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 45,
+    "best_step": 8775
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/config.json

@@ -0,0 +1,46 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "config.json",
+    "optim_name": "adamw",
+    "shed_name": "poly_decay",
+    "sched_num_warmup_steps": 45000,
+    "optim_weight_decay": 0.0,
+    "optim_beta_1": 0.9,
+    "optim_beta_2": 0.999,
+    "optim_epsilon": 1e-08,
+    "sched_lr_end": 1e-07,
+    "sched_power": 1.0,
+    "sched_t_max": 200,
+    "sched_last_epoch": -1,
+    "training_result_file": "training_result.json",
+    "check_finished_file": "finished.json",
+    "project": "CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200",
+    "name_postfix": "ema-run0_998-800-600-400-200",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "FrankCCCCC/ddpm_ema_cifar10",
+    "train_dataset": "cifar10",
+    "traj_dataset": "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000",
+    "batch_size": 256,
+    "num_epochs": 80,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 100,
+    "sigma_min": 0.0,
+    "sigma_src": 0.01,
+    "corr_trained_timestep": 800,
+    "ds_size": 50000,
+    "seed": 42,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep0/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep10/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep15/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep15/scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.34.0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep20/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep25/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep30/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep35/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep40/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep45/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep5/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep50/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep55/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep60/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep65/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep70/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep75/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/epochs/ep80/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/finished.json

@@ -0,0 +1,3 @@
+{
+    "Msg": "Training finished."
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    800
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.34.0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 30,
+    "best_step": 5850
+}

cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200/unet/config.json

@@ -0,0 +1,48 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/home/sc3379_cornell_edu/.cache/huggingface/hub/models--google--ddpm-cifar10-32/snapshots/267b167dc01f0e4e61923ea244e8b988f84deb80",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": null,
+  "attn_norm_num_groups": null,
+  "block_out_channels": [
+    128,
+    256,
+    256,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 0,
+  "downsample_type": "conv",
+  "dropout": 0.0,
+  "flip_sin_to_cos": false,
+  "freq_shift": 1,
+  "in_channels": 3,
+  "layers_per_block": 2,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2D",
+  "norm_eps": 1e-06,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "num_train_timesteps": null,
+  "out_channels": 3,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 32,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D"
+  ],
+  "upsample_type": "conv"
+}

cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200/config.json

@@ -0,0 +1,46 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "config.json",
+    "optim_name": "adamw",
+    "shed_name": "poly_decay",
+    "sched_num_warmup_steps": 45000,
+    "optim_weight_decay": 0.0,
+    "optim_beta_1": 0.9,
+    "optim_beta_2": 0.999,
+    "optim_epsilon": 1e-08,
+    "sched_lr_end": 1e-07,
+    "sched_power": 1.0,
+    "sched_t_max": 200,
+    "sched_last_epoch": -1,
+    "training_result_file": "training_result.json",
+    "check_finished_file": "finished.json",
+    "project": "CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200",
+    "name_postfix": "ema-run0_998-800-600-400-200",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "FrankCCCCC/ddpm_ema_cifar10",
+    "train_dataset": "cifar10",
+    "traj_dataset": "data/traj_dataset/TRAJ-DS_PS-DM_DDPM_PSS-1000_SZ-10000",
+    "batch_size": 256,
+    "num_epochs": 80,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 100,
+    "sigma_min": 0.0,
+    "sigma_src": 0.01,
+    "corr_trained_timestep": 998,
+    "ds_size": 50000,
+    "seed": 42,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200/finished.json

@@ -0,0 +1,3 @@
+{
+    "Msg": "Training finished."
+}

cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    998
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200/sample_config.json

@@ -0,0 +1,73 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "saving_strategy": "best_valid_loss",
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "sample_config.json",
+    "dataset_stat_path": "tts_dm/data/init_bias.safetensors",
+    "fid_dims": 2048,
+    "mode_init_puzzle": "init_puzzle",
+    "mode_final_puzzle": "final_puzzle",
+    "mode_intermediate_puzzle": "intermediate_puzzle",
+    "mode_final_samples": "final_samples",
+    "mode_allowed_ctrl_puzzle": "allowed_ctrl_puzzle",
+    "mode_fid": "fid",
+    "mode_save_traj_dataset": "save_traj_dataset",
+    "supported_modes": [
+        "init_puzzle",
+        "final_puzzle",
+        "intermediate_puzzle",
+        "final_samples",
+        "allowed_ctrl_puzzle",
+        "fid",
+        "save_traj_dataset"
+    ],
+    "training_config_key": "training_config",
+    "project": "default",
+    "name_postfix": "",
+    "mode": "fid",
+    "init_bias_dataset_name": "",
+    "predictor_path": "FrankCCCCC/ddpm_ema_cifar10",
+    "predictor_sched": "DM_DDPM",
+    "predictor_num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_path": [
+        "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200",
+        "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/cfm_corr_800_ss0.01_ep80_ema-run0_998-800-600-400-200",
+        "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/cfm_corr_600_ss0.01_ep80_ema-run0_998-800-600-400-200",
+        "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/cfm_corr_400_ss0.01_ep80_ema-run0_998-800-600-400-200"
+    ],
+    "corrector_sched": "FM_EulDis",
+    "corrector_epoch": -1,
+    "corrector_step": -1,
+    "corrector_num_train_timesteps": 1000,
+    "corrector_num_inference_steps": 100,
+    "controllor_path": "",
+    "result_metric_path": "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/results/results_998-800-600-400_ep-1",
+    "result_sample_path": "fm_cifar10/CFM_CORR_PROG_FINAL_STEP_SIZE200_ema-run0_998-800-600-400-200/results/results_998-800-600-400_ep-1/samples",
+    "traj_path": "",
+    "batch_size": 2500,
+    "sample_num": 10000,
+    "split_sample_num": [
+        [
+            "train",
+            "test"
+        ],
+        [
+            1,
+            1
+        ]
+    ],
+    "seed": 42,
+    "device": "cuda:0",
+    "img_format": "png",
+    "fid_batch_size": 512,
+    "fid_ref_data_path": "data/cifar10_train_png"
+}

cfm_corr_998_ss0.01_ep80_ema-run0_998-800-600-400-200/training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 40,
+    "best_step": 7800
+}

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/test/test-000000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78293ae2edde073fdf9d92ec08a128ac0dae7c218fd36ecb0ac5c2fa38149c52
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/test/test-000001.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3133c629bba1f23203acd6578e15a83689e09efa9e4c22d12bee3d3d6ee39a29
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ea3f216356beb832a282bf1916f89ab5c0e83e2c9254e75b4bdea920ecf7712
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000001.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cd164bec703441e6baea0077c90101bddbae198024e784ed311a5a09ba41c97
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c80a39c2824a89cff842561af8909f0d65d3c51f38beddb1f4394329f01eed7
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa60e1ba72118025c0c5895010c826fb5a536bcd45c83f3aa6ea61a82a8cab33
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc4b86ff3028496190bc234820748426c4cf8b269b4a447965e9c83c8225126a
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f67e01b30075c0161f4ddaf7d1195bf210fa51a0dfba2f8a8ab801a50d084d9a
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7073d9d6b14b1aca159f458477417b3523f41f2b1efde67d4b230f0b2c28e8fd
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c277f15e2d4093c8f09173335a3b48bfd123cc4573ff1ad8510855836cd079d
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b942e7f161485083eb5d2855b8ea84740a3d8641f9bca34de6a37a011f49f048
+size 12300380184

traj_dataset/TRAJ-DS_CORR-STEP-998_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000/data/train/train-000009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88fa6b66824bdf9040b81cb2f8ee57f17a677fdf1d7c4375bade4bd3bf5e4bdb
+size 12300380184