{ "dataset_name": "/home/ubuntu/projects/phase-iv-ai/maps/train", "val_dataset_name": "/home/ubuntu/projects/phase-iv-ai/maps/val", "dataset_config_name": null, "model_config_name_or_path": null, "train_data_dir": null, "output_dir": "/media/share/Datasets/diffusers_out/2024-08-02_14-35-03_MLBriefs24_5_conditional_CA_encodedmask", "overwrite_output_dir": false, "cache_dir": null, "resolution": 256, "center_crop": false, "random_flip": false, "train_batch_size": 4, "eval_batch_size": 16, "dataloader_num_workers": 0, "num_epochs": 405, "save_images_epochs": 50, "save_model_epochs": 50, "gradient_accumulation_steps": 1, "learning_rate": 0.0001, "lr_scheduler": "cosine", "lr_warmup_steps": 500, "adam_beta1": 0.95, "adam_beta2": 0.999, "adam_weight_decay": 1e-06, "adam_epsilon": 1e-08, "use_ema": true, "ema_inv_gamma": 1.0, "ema_power": 0.75, "ema_max_decay": 0.9999, "push_to_hub": false, "hub_token": null, "hub_model_id": null, "hub_private_repo": false, "logger": "tensorboard", "logging_dir": "/media/share/Datasets/diffusers_out/2024-08-02_14-35-03_MLBriefs24_5_conditional_CA_encodedmask", "local_rank": -1, "mixed_precision": "no", "prediction_type": "epsilon", "ddpm_num_steps": 1000, "ddpm_num_inference_steps": 1000, "ddpm_beta_schedule": "linear", "checkpointing_steps": 5000, "checkpoints_total_limit": 5, "resume_from_checkpoint": null, "enable_xformers_memory_efficient_attention": false, "acc_seed": null, "train_data_files": null, "encode_cond": true, "unconditional": false, "crossattention": true }