ZeroCool94 commited on
Commit
583cbe9
1 Parent(s): fbeb4eb

Upload 2 files

Browse files
Files changed (2) hide show
  1. vae.15085000.pt +3 -0
  2. vae.15085000.pt.yaml +67 -0
vae.15085000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b55ad66e2ece3dc1484ea2913a6060a359599bc21ad89e828ea146315dc60c6
3
+ size 577198907
vae.15085000.pt.yaml ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ total_params: 10067428
2
+ image_size: 512
3
+ num_train_steps: -1
4
+ num_epochs: 50
5
+ batch_size: 1
6
+ lr: 0.0001
7
+ lr_warmup_steps: 10
8
+ lr_scheduler: cosine_with_restarts
9
+ gradient_accumulation_steps: 2000
10
+ save_results_every: 500
11
+ save_model_every: 5000
12
+ dim: 32
13
+ vq_codebook_size: 8192
14
+ vq_codebook_dim: 8192
15
+ seq_len: 1024
16
+ channels: 3
17
+ layers: 4
18
+ discr_layers: 4
19
+ scheduler_power: 8.0
20
+ num_cycles: 800
21
+ only_save_last_checkpoint: false
22
+ validation_image_scale: 1.0
23
+ no_center_crop: false
24
+ no_flip: false
25
+ random_crop: true
26
+ dataset_save_path: E:/cached_datasets/INE
27
+ clear_previous_experiments: false
28
+ max_grad_norm: null
29
+ discr_max_grad_norm: null
30
+ seed: 42
31
+ valid_frac: 0.05
32
+ use_ema: false
33
+ ema_beta: 0.995
34
+ ema_update_after_step: 1
35
+ ema_update_every: 1
36
+ apply_grad_penalty_every: 4
37
+ image_column: image
38
+ caption_column: caption
39
+ log_with: wandb
40
+ mixed_precision: 'no'
41
+ use_8bit_adam: false
42
+ results_dir: results\Muse-v0.9_test
43
+ logging_dir: null
44
+ resume_path: results\Muse-v0.9_test\vae.13860000.pt
45
+ dataset_name: null
46
+ streaming: false
47
+ train_data_dir: F:/Hydrus Files
48
+ checkpoint_limit: null
49
+ cond_drop_prob: 0.5
50
+ taming_model_path: null
51
+ taming_config_path: null
52
+ optimizer: Adam
53
+ weight_decay: 0.001
54
+ cache_path: null
55
+ no_cache: true
56
+ latest_checkpoint: true
57
+ do_not_save_config: false
58
+ use_l2_recon_loss: false
59
+ debug: false
60
+ config_path: null
61
+ validation_folder_at_end_of_epoch: samples/validation
62
+ exclude_folders: null
63
+ webdataset: null
64
+ project_name: muse_vae
65
+ run_name: null
66
+ wandb_user: null
67
+ hf_split_name: train