Add files using upload-large-folder tool

README.md

@@ -0,0 +1,22 @@
+# cfm_corr_2_ss0.0_ep100_ema-run1
+
+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_2_ss0.0_ep100_ema-run1
+- Images excluded from upload
+
+## Usage
+
+This repository contains the necessary files to reproduce or analyze the model from this specific experimental run.

ckpt/data.ckpt

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

ckpt/model.safetensors

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

ckpt/optimizer.bin

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

ckpt/random_states_0.pkl

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

ckpt/scheduler.bin

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

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_EMA_Final",
+    "name_postfix": "ema-run1",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "google/ddpm-cifar10-32",
+    "train_dataset": "cifar10",
+    "traj_dataset": "data/traj_dataset/TRAJ-DS_PS-DM_DDPM_PSS-1000_SZ-10000",
+    "batch_size": 256,
+    "num_epochs": 100,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 5,
+    "sigma_min": 0.0,
+    "sigma_src": 0.0,
+    "corr_trained_timestep": 2,
+    "ds_size": 50000,
+    "seed": 42,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

epochs/ep0/model_index.json

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

epochs/ep10/model_index.json

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

epochs/ep10/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
+}

epochs/ep10/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"
+}

epochs/ep100/model_index.json

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

epochs/ep15/model_index.json

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

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
+}

epochs/ep15/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"
+}

epochs/ep15/unet/diffusion_pytorch_model.safetensors

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

epochs/ep20/model_index.json

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

epochs/ep20/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
+}

epochs/ep20/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"
+}

epochs/ep25/model_index.json

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

epochs/ep30/model_index.json

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

epochs/ep35/model_index.json

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

epochs/ep40/model_index.json

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

epochs/ep45/model_index.json

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

epochs/ep5/model_index.json

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

epochs/ep50/model_index.json

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

epochs/ep55/model_index.json

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

epochs/ep60/model_index.json

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

epochs/ep60/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
+}

epochs/ep60/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"
+}

epochs/ep60/unet/diffusion_pytorch_model.safetensors

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

epochs/ep65/model_index.json

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

epochs/ep70/model_index.json

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

epochs/ep70/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
+}

epochs/ep70/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"
+}

epochs/ep75/model_index.json

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

epochs/ep80/model_index.json

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

epochs/ep85/model_index.json

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

epochs/ep90/model_index.json

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

epochs/ep90/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
+}

epochs/ep90/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"
+}

epochs/ep95/model_index.json

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

finished.json

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

model_index.json

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

result.json

@@ -0,0 +1,25 @@
+{
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 9.160543897973866,
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-10_SZ-10000_F-png": 9.246251328038227,
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-1_SZ-10000_F-png": 184.04142366015301,
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-20_SZ-10000_F-png": 9.342601124556609,
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-50_SZ-10000_F-png": 9.295757357548155,
+    "corr_M-fid_EP--1_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-5_SZ-10000_F-png": 9.014279702758245,
+    "corr_M-fid_EP-10_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 8.652539129082243,
+    "corr_M-fid_EP-10_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-10_SZ-10000_F-png": 9.92214611117862,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 7.371318271139387,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-10_SZ-10000_F-png": 8.163039539490285,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-1_SZ-10000_F-png": 118.92836376370201,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-20_SZ-10000_F-png": 7.619706494956631,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-50_SZ-10000_F-png": 7.427592071216623,
+    "corr_M-fid_EP-30_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-5_SZ-10000_F-png": 8.180700282743373,
+    "corr_M-fid_EP-50_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 8.177825406836178,
+    "corr_M-fid_EP-50_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-20_SZ-10000_F-png": 8.68803560560275,
+    "corr_M-fid_EP-5_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 12.569175078986405,
+    "corr_M-fid_EP-5_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-10_SZ-10000_F-png": 14.27274628089333,
+    "corr_M-fid_EP-5_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-20_SZ-10000_F-png": 13.185705580402043,
+    "corr_M-fid_EP-5_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-5_SZ-10000_F-png": 16.8036665707844,
+    "corr_M-fid_EP-80_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-100_SZ-10000_F-png": 8.778016231373897,
+    "corr_M-fid_EP-80_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-20_SZ-10000_F-png": 9.405777085220961,
+    "corr_M-fid_EP-80_PS-DM_DDPM_PSS-1000_CS-FM_EulDis_CSS-5_SZ-10000_F-png": 9.83976431088513
+}

sample_config.json

@@ -0,0 +1,68 @@
+{
+    "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_EMA_Final/cfm_corr_2_ss0.0_ep100_ema-run1",
+    "corrector_sched": "FM_EulDis",
+    "corrector_epoch": 30,
+    "corrector_step": -1,
+    "corrector_num_train_timesteps": 1000,
+    "corrector_num_inference_steps": 50,
+    "controllor_path": "",
+    "result_metric_path": "",
+    "result_sample_path": "",
+    "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"
+}

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
+}

training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 100,
+    "best_step": 19531
+}

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"
+}

unet/diffusion_pytorch_model.safetensors

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