jwheo commited on
Commit
cd65e8e
·
verified ·
1 Parent(s): b1137ce

Upload guarded-detail Stage2 Colab default

Browse files
configs/hf/latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2.yaml ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2_hf
3
+
4
+ autoencoder:
5
+ config: configs/autoencoder_photo10k.yaml
6
+ checkpoint: checkpoints/stage1_autoencoder_best_eval_recon.pt
7
+
8
+ data:
9
+ hr_size: 512
10
+ scale: 4
11
+ degradation_preset: photo_detail_mix
12
+ domains:
13
+ photo: 0
14
+ anime: 1
15
+
16
+ model:
17
+ type: lr_to_latent_predictor
18
+ architecture: dual_multiscale_context
19
+ in_channels: 3
20
+ latent_channels: 16
21
+ base_channels: 256
22
+ num_blocks: 16
23
+ norm_groups: 32
24
+ num_domains: 2
25
+ context_channels: [320, 448]
26
+ context_blocks: [4, 6]
27
+ context_scale: 1.0
28
+ extra_context_channels: [384, 512]
29
+ extra_context_blocks: [6, 8]
30
+ extra_context_scale: 1.0
31
+
32
+ train:
33
+ dtype: bf16
34
+
35
+ inference:
36
+ checkpoint: checkpoints/stage2_photo130k_lsdir_dual_detail_guarded_v2_best10000.pt