| webdataset: null |
| only_save_last_checkpoint: false |
| validation_image_scale: 1.0 |
| no_center_crop: false |
| random_crop: true |
| no_flip: false |
| dataset_save_path: E:/cached_datasets/INE |
| clear_previous_experiments: false |
| max_grad_norm: null |
| discr_max_grad_norm: null |
| seed: 42 |
| valid_frac: 0.05 |
| use_ema: false |
| ema_beta: 0.995 |
| ema_update_after_step: 1 |
| ema_update_every: 1 |
| apply_grad_penalty_every: 4 |
| image_column: image |
| caption_column: caption |
| log_with: wandb |
| mixed_precision: fp16 |
| use_8bit_adam: false |
| results_dir: results\Muse-v0.7 |
| logging_dir: results\Muse-v0.7/logs |
| dataset_name: null |
| train_data_dir: E:\dataset |
| num_train_steps: 2000001 |
| dim: 128 |
| batch_size: 1 |
| lr: 1.0e-05 |
| gradient_accumulation_steps: 20 |
| save_results_every: 100 |
| save_model_every: 500 |
| vq_codebook_size: 16384 |
| vq_codebook_dim: 4096 |
| image_size: 384 |
| lr_scheduler: constant_with_warmup |
| lr_warmup_steps: 100 |
| num_cycles: 1 |
| scheduler_power: 0.5 |
| resume_path: '' |
| optimizer: AdamW |
| weight_decay: 1.0e-05 |
| taming_model_path: null |
| taming_config_path: null |
| use_profiling: false |
| no_cache: true |
| profile_frequency: 1 |
| row_limit: 10 |
| gpu: 0 |
| latest_checkpoint: false |
| do_not_save_config: false |
| use_l2_recon_loss: false |
|
|
