File size: 1,624 Bytes
a7fef13 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | {
"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
} |