Phips commited on
Commit
6cd3ca0
1 Parent(s): 17c7b5e

Upload 27 files

Browse files
Files changed (28) hide show
  1. .gitattributes +7 -0
  2. baseline/DF2K/4xDF2K_plksr_tiny/4xDF2K_plksr_tiny.toml +57 -0
  3. baseline/DF2K/4xDF2K_plksr_tiny/models/net_g_500000.pth +3 -0
  4. baseline/DF2K/4xDF2K_plksr_tiny/train_4xDF2K_plksr_tiny_20240926_193309.log +0 -0
  5. baseline/DF2K/4xDF2K_plksr_tiny/training_states/500000.state +3 -0
  6. baseline/DF2K/4xDF2K_plksr_tiny_bf16/4xDF2K_plksr_tiny_bf16.toml +59 -0
  7. baseline/DF2K/4xDF2K_plksr_tiny_bf16/models/net_g_200000.pth +3 -0
  8. baseline/DF2K/4xDF2K_plksr_tiny_bf16/train_4xDF2K_plksr_tiny_bf16_20241001_035536.log +0 -0
  9. baseline/DF2K/4xDF2K_plksr_tiny_bf16/training_states/200000.state +3 -0
  10. baseline/DF2K/4xDF2K_plksr_tiny_fp16/4xDF2K_plksr_tiny_fp16.toml +58 -0
  11. baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_100000.pth +3 -0
  12. baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_200000.pth +3 -0
  13. baseline/DF2K/4xDF2K_plksr_tiny_fp16/train_4xDF2K_plksr_tiny_fp16_20241001_002907.log +0 -0
  14. baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/100000.state +3 -0
  15. baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/200000.state +3 -0
  16. baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/4xDF2K_plksr_tiny_fp16_full.toml +58 -0
  17. baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/models/net_g_500000.pth +3 -0
  18. baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/train_4xDF2K_plksr_tiny_fp16_full_20241009_171731.log +0 -0
  19. baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/training_states/500000.state +3 -0
  20. baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/4xDF2K_plksr_tiny_fp16_multiscaled.toml +58 -0
  21. baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/models/net_g_200000.pth +3 -0
  22. baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/train_4xDF2K_plksr_tiny_fp16_mutliscaled_20241002_000753.log +0 -0
  23. baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/training_states/200000.state +3 -0
  24. baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/4xDF2K_plksr_tiny_fp16_patch64.toml +58 -0
  25. baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/models/net_g_200000.pth +3 -0
  26. baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_033408.log +0 -0
  27. baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_082607.log +0 -0
  28. baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/training_states/200000.state +3 -0
.gitattributes CHANGED
@@ -1969,3 +1969,10 @@ ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/va
1969
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_098.png filter=lfs diff=lfs merge=lfs -text
1970
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_099.png filter=lfs diff=lfs merge=lfs -text
1971
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_100.png filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
1969
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_098.png filter=lfs diff=lfs merge=lfs -text
1970
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_099.png filter=lfs diff=lfs merge=lfs -text
1971
  ic9600/visualization/urban100/4xIC9600_0.7_DF2K_plksr_tiny_fp16_100k_Urban100/val_1/img_100.png filter=lfs diff=lfs merge=lfs -text
1972
+ baseline/DF2K/4xDF2K_plksr_tiny_bf16/training_states/200000.state filter=lfs diff=lfs merge=lfs -text
1973
+ baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/training_states/500000.state filter=lfs diff=lfs merge=lfs -text
1974
+ baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/training_states/200000.state filter=lfs diff=lfs merge=lfs -text
1975
+ baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/training_states/200000.state filter=lfs diff=lfs merge=lfs -text
1976
+ baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/100000.state filter=lfs diff=lfs merge=lfs -text
1977
+ baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/200000.state filter=lfs diff=lfs merge=lfs -text
1978
+ baseline/DF2K/4xDF2K_plksr_tiny/training_states/500000.state filter=lfs diff=lfs merge=lfs -text
baseline/DF2K/4xDF2K_plksr_tiny/4xDF2K_plksr_tiny.toml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny"
2
+ model_type = "image"
3
+ scale = 4
4
+ manual_seed = 1024
5
+
6
+ [datasets.train]
7
+ type = "paired"
8
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
9
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
10
+ patch_size = 32
11
+ batch_size = 16
12
+
13
+ [datasets.val]
14
+ name = "val"
15
+ type = "paired"
16
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
17
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
18
+ [val]
19
+ val_freq = 10000
20
+ save_img = false
21
+ [val.metrics.psnr]
22
+ type = "calculate_psnr"
23
+ [val.metrics.ssim]
24
+ type = "calculate_ssim"
25
+ [val.metrics.dists]
26
+ type = "calculate_dists"
27
+ better = "lower"
28
+
29
+ [path]
30
+
31
+ [network_g]
32
+ #type = "plksr"
33
+ type = "plksr_tiny"
34
+
35
+ [train]
36
+ ema = 0.999
37
+
38
+ [train.optim_g]
39
+ type = "adamw"
40
+ lr = 1e-4
41
+ betas = [ 0.9, 0.99 ]
42
+ weight_decay = 0
43
+
44
+ [train.scheduler]
45
+ type = "multisteplr"
46
+ milestones = [ 60000, 120000 ]
47
+ gamma = 0.5
48
+
49
+ [train.pixel_opt]
50
+ type = "L1Loss"
51
+ loss_weight = 1.0
52
+ reduction = "mean"
53
+
54
+ [logger]
55
+ total_iter = 500000
56
+ save_checkpoint_freq = 10000
57
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny/models/net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccfef467d9603724f3bcbf938ad01d7560ef162421aeaa9dfea183c9b7a1a676
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny/train_4xDF2K_plksr_tiny_20240926_193309.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny/training_states/500000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de125d6476b28e0f8f500d3f8d6dd38d8399d8836c92bfa3f8673e0172f9a671
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_bf16/4xDF2K_plksr_tiny_bf16.toml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_bf16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ bfloat16 = true
6
+ manual_seed = 1024
7
+
8
+ [datasets.train]
9
+ type = "paired"
10
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
11
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
12
+ patch_size = 32
13
+ batch_size = 16
14
+
15
+ [datasets.val]
16
+ name = "val"
17
+ type = "paired"
18
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
19
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
20
+ [val]
21
+ val_freq = 10000
22
+ save_img = false
23
+ [val.metrics.psnr]
24
+ type = "calculate_psnr"
25
+ [val.metrics.ssim]
26
+ type = "calculate_ssim"
27
+ [val.metrics.dists]
28
+ type = "calculate_dists"
29
+ better = "lower"
30
+
31
+ [path]
32
+
33
+ [network_g]
34
+ #type = "plksr"
35
+ type = "plksr_tiny"
36
+
37
+ [train]
38
+ ema = 0.999
39
+
40
+ [train.optim_g]
41
+ type = "adamw"
42
+ lr = 1e-4
43
+ betas = [ 0.9, 0.99 ]
44
+ weight_decay = 0
45
+
46
+ [train.scheduler]
47
+ type = "multisteplr"
48
+ milestones = [ 60000, 120000 ]
49
+ gamma = 0.5
50
+
51
+ [train.pixel_opt]
52
+ type = "L1Loss"
53
+ loss_weight = 1.0
54
+ reduction = "mean"
55
+
56
+ [logger]
57
+ total_iter = 200000
58
+ save_checkpoint_freq = 10000
59
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny_bf16/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df446e3520a58cb039fd7c107e991cc2faf1f4ef6e05bc0e2fb3e4231cb320ad
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_bf16/train_4xDF2K_plksr_tiny_bf16_20241001_035536.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_bf16/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c9af2c4e23b026ae8963f802ff61b0a13a38fc54b8dec90a3656822605733ae
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_fp16/4xDF2K_plksr_tiny_fp16.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_100000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd0e027df7bc6e3b7c5b901f1aa1e41130fc5b8b796cf9ad1b73b9c9af0c62f3
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6d1560850df99e656c4cf04e15a00fc4e74ed9d3e4ea11b7f24ff7a4227f872
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_fp16/train_4xDF2K_plksr_tiny_fp16_20241001_002907.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/100000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f97e9d96d051ab4723ffcca5a3ef224465d6d3e0781670487f6de47a3383779d
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:854d12454ba1139b511da43cf1c310f3b77139f7c4d51169a9de8942efe698cf
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/4xDF2K_plksr_tiny_fp16_full.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_full"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 500000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/models/net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b3147f206c25a52fd922ad8cfc85be8da3cc760ad5ac135e70444cb701b0683
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/train_4xDF2K_plksr_tiny_fp16_full_20241009_171731.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/training_states/500000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:306638a02f6869135eaf7298409d4c310c08d717c83a3c65e73ccac50191a9a2
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/4xDF2K_plksr_tiny_fp16_multiscaled.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_mutliscaled"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_multiscaled_tiled_merged_HR/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_multiscaled_tiled_merged_LR/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9cbb26b3404ece32a1ebcd163d56502644105846a11cf72db20acbfe799d205
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/train_4xDF2K_plksr_tiny_fp16_mutliscaled_20241002_000753.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63f41bb98712163a9b6fdf51192cb1c0f16f3bfcd0450cd6daa0a803ffc706c4
3
+ size 19045418
baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/4xDF2K_plksr_tiny_fp16_patch64.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_patch64"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 64
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:682f0143c2c8c2d36e608591ff43436bde79a161d90c26aff6f76625a3d98d9a
3
+ size 9513138
baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_033408.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_082607.log ADDED
The diff for this file is too large to render. See raw diff
 
baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a63a9ae8e1983372f0bffd02176928db53106bb9bda152c300d11fff68a15522
3
+ size 19045418