jwheo commited on
Commit
e651489
·
verified ·
1 Parent(s): 52653e2

Upload photo100k Stage 3 v2 diffusion checkpoint

Browse files
metrics/stage3_photo100k_v2_b32_summary.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_decoded_psnr": {
3
+ "eval/decoded_mse": 0.018126084357500075,
4
+ "eval/decoded_psnr": 23.43755994490881,
5
+ "eval/noise_mse": 0.00821230050176382,
6
+ "eval/num_images": 100.0,
7
+ "eval/x0_mse": 0.09724276781082153,
8
+ "step": 19000
9
+ },
10
+ "best_noise_mse": {
11
+ "eval/decoded_mse": 0.018126084357500075,
12
+ "eval/decoded_psnr": 23.43755994490881,
13
+ "eval/noise_mse": 0.00821230050176382,
14
+ "eval/num_images": 100.0,
15
+ "eval/x0_mse": 0.09724276781082153,
16
+ "step": 19000
17
+ },
18
+ "checkpoint": "checkpoints/stage3_photo100k_v2_b32_best_eval_noise.pt",
19
+ "condition_encoder": "checkpoints/stage2_photo100k_v2_b64_best_eval_latent.pt",
20
+ "dataset": "manifest_photo100k.csv",
21
+ "degradation_preset": "photo_v2",
22
+ "final": {
23
+ "eval/decoded_mse": 0.018136419355869293,
24
+ "eval/decoded_psnr": 23.435084422027522,
25
+ "eval/noise_mse": 0.00823758251965046,
26
+ "eval/num_images": 100.0,
27
+ "eval/x0_mse": 0.09751181840896607,
28
+ "step": 20000
29
+ },
30
+ "finished_step": 20000,
31
+ "initialized_from": "checkpoints/stage3_photo100k_b32_best_eval_noise.pt",
32
+ "run": "diffusion_photo100k_b32_v2",
33
+ "sampled_eval": {
34
+ "checkpoint": "/home/jwheojjang/scratch/sr-diffusion/runs/diffusion_photo100k_b32_v2/checkpoints/best_eval_noise.pt",
35
+ "checkpoint_step": 19000,
36
+ "config": "configs/diffusion_photo100k_b32_v2.yaml",
37
+ "eta": 0.0,
38
+ "init": "condition",
39
+ "limit": 100,
40
+ "mean_bicubic_psnr": 22.410346417149967,
41
+ "mean_psnr_delta": 0.259511047033417,
42
+ "mean_sr_psnr": 22.669857464183384,
43
+ "split": "val",
44
+ "start_timestep": null,
45
+ "steps": 32
46
+ },
47
+ "train_images": 103450,
48
+ "val_images": 100
49
+ }