_everydream2_config/train.json

{
  "amp": true,
  "batch_size": 6,
  "ckpt_every_n_minutes": null,
  "clip_grad_norm": null,
  "clip_skip": 0,
  "cond_dropout": 0.0,
  "data_root": "/workspace/val-test-7e-8",
  "disable_textenc_training": false,
  "disable_xformers": false,
  "flip_p": 0.0,
  "gpuid": 0,
  "gradient_checkpointing": true,
  "grad_accum": 3,
  "logdir": "/workspace/logs",
  "log_step": 25,
  "lowvram": false,
  "lr": 1e-06,
  "lr_decay_steps": 0,
  "lr_scheduler": "constant",
  "lr_warmup_steps": null,
  "max_epochs": 60,
  "notebook": false,
  "project_name": "pashashlis-1e-6",
  "resolution": 768,
  "resume_ckpt": "runwayml/stable-diffusion-v1-5",
  "sample_prompts": "sample_prompts.json",
  "sample_steps": 246,
  "save_ckpt_dir": null,
  "save_ckpts_from_n_epochs": 0,
  "save_every_n_epochs": 5,
  "save_optimizer": false,
  "scale_lr": false,
  "seed": -1,
  "shuffle_tags": false,
  "useadam8bit": true,
  "validation_config": "/everydream2/validation_default.json",
  "wandb": false,
  "write_schedule": false,
  "rated_dataset": false,
  "rated_dataset_target_dropout_percent": 50,
  "zero_frequency_noise_ratio": 0.02
}