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

Upload guarded-detail Stage2 Colab default

Browse files
metrics/stage2_photo130k_lsdir_dual_detail_guarded_v2_best10000_summary.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "checkpoint": "checkpoints/stage2_photo130k_lsdir_dual_detail_guarded_v2_best10000.pt",
3
+ "config": "configs/latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2.yaml",
4
+ "hf_config": "configs/hf/latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2.yaml",
5
+ "run_name": "latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2",
6
+ "wandb_run": "https://wandb.ai/jwheo/LuSIR/runs/fqhftgwf",
7
+ "source_checkpoint": "/home/ubuntu/scratch/sr-diffusion/runs/latent_pretrain_photo130k_lsdir_dual_detail_guarded_v2/checkpoints/best_eval_mean_psnr_detail.pt",
8
+ "selected_step": 10000,
9
+ "final_step": 20000,
10
+ "selection": "best eval/mean_psnr_detail_score; final step 20000 plateaued and was not the best detail checkpoint",
11
+ "params": 119238352,
12
+ "model_weight_size_gb": 0.4442,
13
+ "training_checkpoint_size_gb": 1.4,
14
+ "val100": {
15
+ "decoded_psnr": 24.6296,
16
+ "decoded_mean_psnr": 26.505,
17
+ "highpass_energy_ratio": 0.8084,
18
+ "missing_energy": 0.01897,
19
+ "detail_score_rank": "best"
20
+ },
21
+ "measured_inference_memory_l40s_bf16_128lr": {
22
+ "tile_batch_1_reserved_gb": 1.03,
23
+ "tile_batch_4_reserved_gb": 3.28,
24
+ "tile_batch_8_reserved_gb": 6.25
25
+ },
26
+ "recommended_colab": {
27
+ "gpu": "T4 or better",
28
+ "tile_batch_size": 1,
29
+ "tta_default": "Off",
30
+ "tta_options": ["Horizontal flip x2", "Full x8 self-ensemble"]
31
+ }
32
+ }