AMERICAN666 commited on
Commit
059ff53
·
verified ·
1 Parent(s): aafeb41

Add files using upload-large-folder tool

Browse files
4xNomos2_hq_drct-l_UPSCALE/4xNomos2_hq_drct-l.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f10855fb26dfc2e1c475188d0f16e4692e2e7ca792d46ddf39d75b6ed2326dd
3
+ size 242743038
4xNomos2_hq_drct-l_UPSCALE/4xNomos2_hq_drct-l.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a10632101bb2aca6151301dd811921f1c438ff55159fd579fb576c143477797
3
+ size 242543780
4xNomos2_hq_drct-l_UPSCALE/4xNomos2_hq_drct-l.toml ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xNomos2_hq_drct-l"
2
+ model_type = "image"
3
+ scale = 4
4
+ #use_amp = true # off at 150k
5
+ #bfloat16 = true # off at 150k
6
+ #fast_matmul = true # off at 150k
7
+ #compile = true
8
+ #manual_seed = 1024
9
+
10
+ [datasets.train]
11
+ type = "paired"
12
+ dataroot_gt = '/home/phips/Documents/Datasets/nomosv2/'
13
+ dataroot_lq = '/home/phips/Documents/Datasets/nomosv2_lq_4x/'
14
+ patch_size = 64 # from 32 to 64 at 150k
15
+ batch_size = 2
16
+ #accumulate = 1
17
+ #augmentation = [ "none", "mixup", "cutmix", "resizemix" ] # [ "cutblur" ]
18
+ #aug_prob = [ 0.5, 0.1, 0.1, 0.1 ] # [ 0.7 ]
19
+
20
+ #[datasets.val]
21
+ #name = "val"
22
+ #type = "paired"
23
+ #dataroot_gt = '/home/phips/Documents/Datasets/LSDIR_val/few2_hr/'
24
+ #dataroot_lq = '/home/phips/Documents/Datasets/LSDIR_val/few2_x4/'
25
+ #[val]
26
+ #val_freq = 100
27
+ #tile = 128
28
+ #[val.metrics.psnr]
29
+ #type = "calculate_psnr"
30
+ #[val.metrics.ssim]
31
+ #type = "calculate_ssim"
32
+ #[val.metrics.dists]
33
+ #type = "calculate_dists"
34
+ #better = "lower"
35
+
36
+ [path]
37
+ pretrain_network_g = '/home/phips/Downloads/DRCT-L_X4.pth'
38
+ #pretrain_network_d = 'experiments\pretrain_d.pth'
39
+
40
+ [network_g]
41
+ #type = "drct"
42
+ type = "drct_l"
43
+ #type = "drct_xl"
44
+ #type = "drct_s"
45
+
46
+
47
+ [network_d]
48
+ type = "unet"
49
+
50
+ [train]
51
+ ema = 0.999
52
+ #sam = "fsam"
53
+ #sam_init = 1000
54
+ #eco = true
55
+ #eco_init = 15000
56
+ #wavelet_guided = true
57
+ #wavelet_init = 80000
58
+ #match_lq_colors = true
59
+
60
+ [train.optim_g]
61
+ type = "adamw"
62
+ lr = 1e-4
63
+ betas = [ 0.9, 0.99 ]
64
+ weight_decay = 0
65
+ #schedule_free = true
66
+ #warmup_steps = 1600
67
+
68
+ [train.optim_d]
69
+ type = "adamw"
70
+ lr = 1e-4
71
+ betas = [ 0.9, 0.99 ]
72
+ weight_decay = 0
73
+ #schedule_free = true
74
+ #warmup_steps = 600
75
+
76
+ # losses
77
+ [train.mssim_opt]
78
+ type = "mssim_loss"
79
+ loss_weight = 1.0
80
+
81
+ [train.consistency_opt]
82
+ type = "consistency_loss"
83
+ loss_weight = 1.0
84
+ #saturation = 1.0
85
+ #brightness = 1.0
86
+
87
+ [train.perceptual_opt]
88
+ type = "vgg_perceptual_loss"
89
+ loss_weight = 0.5
90
+ criterion = "chc"
91
+ #patchloss = true
92
+ #ipk = true
93
+ #patch_weight = 1.0
94
+
95
+ [train.gan_opt]
96
+ type = "gan_loss"
97
+ gan_type = "bce"
98
+ loss_weight = 0.3
99
+
100
+ [train.dists_opt] #enable at 172k
101
+ type = "dists_loss"
102
+ loss_weight = 0.5
103
+
104
+ [train.ldl_opt] #enable at 172k
105
+ type = "ldl_loss"
106
+ loss_weight = 1.0
107
+ criterion = "chc"
108
+
109
+ [train.ff_opt] #enable at 172k
110
+ type = "ff_loss"
111
+ loss_weight = 1.0
112
+
113
+ #[train.gw_opt]
114
+ #type = "gw_loss"
115
+ #loss_weight = 1.0
116
+
117
+ [logger]
118
+ total_iter = 1000000
119
+ save_checkpoint_freq = 5000
120
+ use_tb_logger = true
121
+ #save_tb_img = true
122
+ #print_freq = 100