Upload 2 files

vae.87000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31323e9061ed3488e1eab98d3cc483174e9ed5e9653519272a956838d475f9a9
+size 676691876

vae.87000.pt.yaml

@@ -0,0 +1,53 @@
+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