| { | |
| "beta_1": 0.9, | |
| "beta_2": 0.999, | |
| "epsilon": 1e-08, | |
| "lr_sched_num_warmup_steps": 45000, | |
| "lr_sched_lr_end": 1e-07, | |
| "lr_sched_power": 1.0, | |
| "ep_model_dir": "epochs", | |
| "output_dir": "fm_cifar10", | |
| "ckpt_dir": "ckpt", | |
| "data_ckpt_dir": "data.ckpt", | |
| "is_save_all_model_epochs": false, | |
| "args_key": "args", | |
| "default_key": "default", | |
| "final_key": "final", | |
| "config_file": "config.json", | |
| "project": "cfm-training", | |
| "run_name": "train_cfm", | |
| "model_id": "google/ddpm-cifar10-32", | |
| "batch_size": 256, | |
| "num_epochs": 1000, | |
| "lr": 0.0005, | |
| "weight_decay": 0.0, | |
| "num_train_timesteps": 1000, | |
| "num_inference_steps": 1000, | |
| "sigma_min": 0.0, | |
| "seed": 42, | |
| "device": "cuda:0" | |
| } |