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