| { |
| "adam_betas": "0.9,0.99", |
| "amp": true, |
| "attn_drop": 0.0, |
| "auto_normalize": false, |
| "beta_schedule": "cosine", |
| "comb_method": "add", |
| "compile_model": false, |
| "cond_drop_prob": 0.2, |
| "config": null, |
| "crop": true, |
| "crop_size": 32, |
| "data_channels": 16, |
| "dataloader_num_workers": 4, |
| "dataloader_persistent_workers": true, |
| "dataloader_prefetch_factor": 2, |
| "dataset_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_dir", |
| "ddim_sampling_eta": 0.0, |
| "default_cond_scale": 4.0, |
| "depth": 25, |
| "device": "cuda", |
| "ema_decay": 0.995, |
| "ema_update_every": 10, |
| "embeddings_dict_key": null, |
| "embeddings_path": "assets/full_embeddings_16_raw.pt", |
| "evaluate_ema_model": false, |
| "exp_name": "DDPM_1e4_balanced", |
| "gradient_accumulate_every": 4, |
| "hidden_channels": 768, |
| "image_size": 32, |
| "mappings_file_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_oh_mappings.pt", |
| "max_grad_norm": 1.0, |
| "min_lr_ratio": 0.0, |
| "min_snr_gamma": 5.0, |
| "min_snr_loss_weight": true, |
| "mixed_precision_type": "bf16", |
| "mlp_ratio": 4.0, |
| "num_classes": 15, |
| "num_heads": 8, |
| "num_samples": 15, |
| "objective": "pred_v", |
| "offset_noise_strength": 0.0, |
| "optimizer": "Adam", |
| "patch_size": 2, |
| "per_device_train_batch_size": 32, |
| "proj_drop": 0.0, |
| "results_folder": "./results", |
| "rotation_aug_prob": 0.75, |
| "run_name": "DDPM_crop_32_full_balanced_dataset", |
| "sampling_timesteps": 1000, |
| "save_and_sample_every": 3514, |
| "save_and_sample_every_epoch": 2, |
| "save_only_last_checkpoint": true, |
| "scheduler": "cosine", |
| "split_batches": false, |
| "time_dim": 256, |
| "timesteps": 1000, |
| "train_batch_size": 512, |
| "train_lr": 0.001, |
| "train_num_epochs": 20, |
| "train_num_steps": 35140, |
| "use_wandb": true, |
| "val_dataset_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_val_dir", |
| "val_mappings_file_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_oh_mappings.pt", |
| "validate_every": 3514, |
| "validate_every_epoch": 2, |
| "wandb_group_name": "DDPM", |
| "wandb_log_images": true, |
| "wandb_project": "dream-cubed", |
| "wandb_run_name": "DDPM_crop_32_full_balanced_dataset", |
| "warmup_steps": 0, |
| "weight_decay": 0.01 |
| } |
