sample_config.json

{
    "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_50_ss0.0_ep100_ema-run1",
    "corrector_sched": "FM_EulDis",
    "corrector_epoch": 100,
    "corrector_step": -1,
    "corrector_num_train_timesteps": 1000,
    "corrector_num_inference_steps": 1000,
    "controllor_path": "",
    "result_metric_path": "",
    "result_sample_path": "",
    "traj_path": "",
    "batch_size": 2500,
    "sample_num": 10000,
    "split_sample_num": [
        [
            "train",
            "test"
        ],
        [
            1,
            1
        ]
    ],
    "seed": 43,
    "device": "cuda:0",
    "img_format": "png",
    "fid_batch_size": 512,
    "fid_ref_data_path": "data/cifar10_train_png"
}