args.json

{
  "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
}