jwheo commited on
Commit
e2cd87b
·
verified ·
1 Parent(s): 91d3659

Upload decoded-detail residual refiner v2 artifacts

Browse files
configs/residual_refiner_stage2_xl_photo_detail_v2_long.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: residual_refiner_stage2_xl_photo_detail_v2_long
3
+ output_dir: /home/jwheojjang/scratch/sr-diffusion/runs/residual_refiner_stage2_xl_photo_detail_v2_long
4
+
5
+ seed: 1337
6
+
7
+ autoencoder:
8
+ config: configs/autoencoder_photo10k.yaml
9
+ checkpoint: /home/jwheojjang/scratch/sr-diffusion/runs/autoencoder_photo10k_b16_eval_online/checkpoints/best_eval_recon.pt
10
+
11
+ condition_encoder:
12
+ config: configs/latent_pretrain_photo100k_v3_noise_xl.yaml
13
+ checkpoint: /home/jwheojjang/scratch/sr-diffusion/runs/latent_pretrain_photo100k_v3_noise_xl_b64/checkpoints/step_0072000.pt
14
+ trainable: false
15
+
16
+ data:
17
+ manifest: /home/jwheojjang/scratch/sr-diffusion/data/manifest_photo100k.csv
18
+ split: train
19
+ hr_size: 512
20
+ scale: 4
21
+ degradation_preset: photo_detail_mix
22
+ domains:
23
+ photo: 0
24
+ anime: 1
25
+ num_workers: 4
26
+
27
+ model:
28
+ latent_channels: 16
29
+ lr_channels: 3
30
+ hidden_channels: 192
31
+ num_blocks: 12
32
+ norm_groups: 32
33
+ num_domains: 2
34
+ residual_scale: 1.25
35
+ gate_bias: 0.0
36
+
37
+ loss:
38
+ latent_weight: 1.0
39
+ residual_weight: 0.5
40
+ highpass_weight: 1.5
41
+ decoded_weight: 0.1
42
+ decoded_highpass_weight: 0.5
43
+ gate_l1_weight: 0.0005
44
+ highpass_kernel: 15
45
+ decoded_highpass_kernel: 15
46
+ charbonnier_eps: 0.001
47
+
48
+ train:
49
+ device: auto
50
+ dtype: bf16
51
+ batch_size: 12
52
+ max_steps: 12000
53
+ lr: 0.00005
54
+ weight_decay: 0.0
55
+ grad_accum_steps: 2
56
+ log_every: 25
57
+ save_every: 2000
58
+
59
+ eval:
60
+ enabled: true
61
+ split: val
62
+ limit: 100
63
+ batch_size: 8
64
+ num_workers: 4
65
+ every: 500
66
+ run_at_start: true
67
+ best_metric: eval/refined_decoded_psnr
68
+ best_mode: max
69
+ sample_count: 8