jwheo commited on
Commit
86cbd04
·
verified ·
1 Parent(s): 7627a1b

Add photo100k XL configs

Browse files
configs/latent_pretrain_photo100k_v3_noise_xl.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: latent_pretrain_photo100k_v3_noise_xl_b64
3
+ output_dir: /home/jwheojjang/scratch/sr-diffusion/runs/latent_pretrain_photo100k_v3_noise_xl_b64
4
+
5
+ logging:
6
+ wandb:
7
+ enabled: true
8
+ project: sr-diffusion
9
+ name: latent_pretrain_photo100k_v3_noise_xl_b64
10
+ mode: online
11
+ dir: /home/jwheojjang/scratch/sr-diffusion/wandb
12
+ tags: [stage2, latent-pretrain, photo100k, degradation-v3-noise, xl, condition-encoder-19m, b64]
13
+ watch: false
14
+ samples:
15
+ enabled: true
16
+ split: val
17
+ fallback_split: train
18
+ count: 4
19
+ indices: [0, 1, 2, 3]
20
+
21
+ seed: 1337
22
+
23
+ autoencoder:
24
+ config: configs/autoencoder_photo10k.yaml
25
+ checkpoint: /home/jwheojjang/scratch/sr-diffusion/runs/autoencoder_photo10k_b16_eval_online/checkpoints/best_eval_recon.pt
26
+
27
+ data:
28
+ manifest: /home/jwheojjang/scratch/sr-diffusion/data/manifest_photo100k.csv
29
+ split: train
30
+ hr_size: 512
31
+ scale: 4
32
+ degradation_preset: photo_v3_noise_mix
33
+ domains:
34
+ photo: 0
35
+ anime: 1
36
+ num_workers: 8
37
+
38
+ model:
39
+ type: lr_to_latent_predictor
40
+ in_channels: 3
41
+ latent_channels: 16
42
+ base_channels: 256
43
+ num_blocks: 16
44
+ norm_groups: 32
45
+ num_domains: 2
46
+
47
+ train:
48
+ device: auto
49
+ dtype: bf16
50
+ batch_size: 64
51
+ max_steps: 80000
52
+ lr: 0.00003
53
+ weight_decay: 0.0
54
+ grad_accum_steps: 1
55
+ log_every: 50
56
+ save_every: 1000
57
+ sample_every: 1000
58
+
59
+ eval:
60
+ enabled: true
61
+ split: val
62
+ limit: 100
63
+ batch_size: 32
64
+ num_workers: 8
65
+ every: 1000
66
+ run_at_start: true
67
+
68
+ loss:
69
+ latent: charbonnier