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
|