| { | |
| "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5", | |
| "revision": null, | |
| "tokenizer_name": null, | |
| "instance_data_dir": "data/FFHQ_last2000_512", | |
| "class_data_dir": null, | |
| "instance_prompt": "photo of a person", | |
| "class_prompt": null, | |
| "with_prior_preservation": false, | |
| "with_prior_preservation_attn": false, | |
| "with_prior_preservation_lpips": false, | |
| "with_prior_preservation_unet": false, | |
| "type_prior_preservation_unet": "SD-SD", | |
| "with_prior_preservation_clip": false, | |
| "type_prior_preservation_clip": "img-img", | |
| "with_prior_preservation_sks": false, | |
| "prior_preservation_sks_step": 50, | |
| "prior_loss_weight": 1.0, | |
| "prior_attn_loss_weight": 0, | |
| "prior_lpips_loss_weight": 0, | |
| "prior_unet_loss_weight": 0, | |
| "prior_clip_loss_weight": 0, | |
| "prior_sks_loss_weight": 0, | |
| "num_class_images": 100, | |
| "output_dir": "./saved_model/age/0427_evaluation/finetune_double_prompt_150_random", | |
| "seed": null, | |
| "resolution": 512, | |
| "center_crop": false, | |
| "train_text_encoder": false, | |
| "train_batch_size": 1, | |
| "sample_batch_size": 4, | |
| "num_train_epochs": 1, | |
| "max_train_steps": 150, | |
| "checkpointing_steps": 500, | |
| "resume_from_checkpoint": null, | |
| "gradient_accumulation_steps": 1, | |
| "gradient_checkpointing": false, | |
| "learning_rate": 5e-06, | |
| "scale_lr": false, | |
| "lr_scheduler": "constant", | |
| "lr_warmup_steps": 500, | |
| "lr_num_cycles": 1, | |
| "lr_power": 1.0, | |
| "use_8bit_adam": false, | |
| "adam_beta1": 0.9, | |
| "adam_beta2": 0.999, | |
| "adam_weight_decay": 0.01, | |
| "adam_epsilon": 1e-08, | |
| "max_grad_norm": 1.0, | |
| "push_to_hub": false, | |
| "hub_token": null, | |
| "hub_model_id": null, | |
| "logging_dir": "logs", | |
| "allow_tf32": false, | |
| "report_to": "tensorboard", | |
| "mixed_precision": null, | |
| "prior_generation_precision": null, | |
| "local_rank": -1, | |
| "enable_xformers_memory_efficient_attention": false, | |
| "multi_prompt": false, | |
| "replace_with_multi_prompt": false, | |
| "instance_age_path": "data/FFHQ_all_ages.npy", | |
| "filenames_to_use": "False" | |
| } |