sunshineatnoon commited on
Commit
2bff7bc
1 Parent(s): 2e1db3f

Add application file

Browse files
weights/.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ *.pth filter=lfs diff=lfs merge=lfs -text
weights/108073/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82deafb20a9d564738b5db3ea4227d56c16fe3dcd6ba16350987ae1b14943234
3
+ size 186192007
weights/108073/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/108073.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/108073",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/108073",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/108073/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/108073 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/108073 [default: None]
44
+ patch_size: 40
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/12003/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c61f11b2987362d958f9c93c7c20f99e06b586c022bc036b9aa78f3ba9ed16e
3
+ size 186192007
weights/12003/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/12003.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/12003",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/12003",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/12003/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/12003 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/12003 [default: None]
44
+ patch_size: 40
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/12074/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b4fd534430ff3a2abecf57c1c27e8a71df97ba0e37ebac972a7198fae8db7f5
3
+ size 186192007
weights/12074/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/12074.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/12074",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/12074",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/12074/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/12074 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/12074 [default: None]
44
+ patch_size: 40
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/134008/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b47346d82b2df88c6a7bb677f593808de14ea3c5087b9f63399e87b290e8422a
3
+ size 186192007
weights/134008/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/134008.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/134008",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/134008",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/134052/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a6db083361caad4307f772d2cf76508ea8e22815f98f4d66c42b129f901f46f
3
+ size 186192007
weights/134052/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/134052.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/134052",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/134052",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/138032/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a77860341bceb548eb434cd6936e5586a6a62ef7c73d5c321ae2637a9090c537
3
+ size 186192007
weights/138032/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/138032.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/138032",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/138032",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/145053/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94706bd93cd68f57055691cf4fbce06f836e85607f8b96b06329e628dd6ccb35
3
+ size 186192007
weights/145053/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/145053.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/145053",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/145053",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/164074/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41786bd1075e7840b03133342ec16f104e0f14e728754cfb2a5927737c946bcd
3
+ size 186192007
weights/164074/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/164074.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/164074",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/164074",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/169012/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15cec08537e35e31c4a593ffd9bc6b7463244d554e573eb2ac00ab89bb1f432c
3
+ size 186192007
weights/169012/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/169012.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/169012",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/169012",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/169012/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/169012 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/169012 [default: None]
44
+ patch_size: 40
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/198023/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ec878f0891c65a4059c47c0bb78289273719edccb284223277fcce0cc042bf8
3
+ size 186192007
weights/198023/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/198023.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/198023",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/198023",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/198023/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/198023 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/198023 [default: None]
44
+ patch_size: 128 [default: 40]
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/25098/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:244031d415c96a17cf0117f416e4643f6960ce8640a2b7e27394f2401f72dc5d
3
+ size 186192007
weights/25098/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/25098.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/25098",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/25098",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/25098/exp_args.txt ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ add_clustering_epoch: 0 [default: 1000]
2
+ add_gcn_epoch: 0 [default: None]
3
+ add_self_loops: 1
4
+ add_texture_epoch: 0 [default: 1000]
5
+ batch_size: 1
6
+ beta: 0.999
7
+ config_file: models/week0417/json/single_scale_grouping_ft.json [default: None]
8
+ crop_size: 224
9
+ data_path: /home/xtli/DATA/BSR_processed/train_extend
10
+ dataset: dataset [default: None]
11
+ dec_input_mode: sine_wave_noise [default: None]
12
+ display_freq: 100
13
+ exp_name: 04-18/25098 [default: None]
14
+ gumbel: 0
15
+ hidden_dim: 256 [default: None]
16
+ img_path: None
17
+ l1_loss_wt: 1.0
18
+ lambda_GAN: 1 [default: None]
19
+ lambda_L1: 1 [default: None]
20
+ lambda_style_loss: 1.0 [default: None]
21
+ local_rank: None
22
+ log_freq: 10
23
+ lr: 5e-05 [default: 0.1]
24
+ lr_decay_freq: 3000
25
+ maxIter: 1000
26
+ model_name: model [default: None]
27
+ momentum: 0.5
28
+ nChannel: 100
29
+ nConv: 2
30
+ n_cluster: 10 [default: None]
31
+ n_layers_D: 3 [default: None]
32
+ nepochs: 100 [default: None]
33
+ netE_nc_steepness: 2.0
34
+ netE_num_downsampling_gl: 2
35
+ netE_num_downsampling_sp: 4
36
+ netE_scale_capacity: 1.0
37
+ netG_num_base_resnet_layers: 2
38
+ netG_resnet_ch: 256
39
+ netG_scale_capacity: 1.0
40
+ no_ganFeat_loss: False
41
+ num_D: 2 [default: None]
42
+ num_classes: 0
43
+ out_dir: /home/xtli/WORKDIR/04-18/25098 [default: None]
44
+ patch_size: 40
45
+ perceptual_loss_wt: 1.0
46
+ pretrained_ae: /home/xtli/WORKDIR/07-16/transformer/cpk.pth
47
+ pretrained_path: /home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth [default: None]
48
+ project_name: test_time
49
+ save_freq: 1000 [default: 2000]
50
+ sine_weight: 1 [default: None]
51
+ sp_num: None
52
+ spatial_code_ch: 8
53
+ spatial_code_dim: 32 [default: 256]
54
+ temperature: 23 [default: 1]
55
+ test_time: 0
56
+ texture_code_ch: 256
57
+ use_slic: True
58
+ use_wandb: False
59
+ work_dir: /home/xtli/WORKDIR
60
+ workers: 4
weights/277095/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec4034575069215b9011b225483e296f931556049356012c1eb6e506954b89ef
3
+ size 186192007
weights/277095/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/277095.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/277095",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/277095",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }
weights/45077/cpk.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9146884b9689062a79db7efecd933ca340ffe451716da96a84223c0fbd1fcf97
3
+ size 186192007
weights/45077/exp_args.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/xtli/DATA/BSR_processed/train",
3
+ "img_path": "data/images/45077.jpg",
4
+ "test_path": null,
5
+ "crop_size": 224,
6
+ "scale_size": null,
7
+ "batch_size": 1,
8
+ "workers": 4,
9
+ "pretrained_path": "/home/xtli/WORKDIR/04-15/single_scale_grouping_resume/cpk.pth",
10
+ "hidden_dim": 256,
11
+ "spatial_code_dim": 32,
12
+ "tex_code_dim": 256,
13
+ "exp_name": "04-18/45077",
14
+ "project_name": "ssn_transformer",
15
+ "nepochs": 20,
16
+ "lr": 5e-05,
17
+ "momentum": 0.5,
18
+ "beta": 0.999,
19
+ "lr_decay_freq": 3000,
20
+ "save_freq": 1000,
21
+ "save_freq_iter": 2000000000000,
22
+ "log_freq": 10,
23
+ "display_freq": 100,
24
+ "use_wandb": 0,
25
+ "work_dir": "/home/xtli/WORKDIR",
26
+ "out_dir": "/home/xtli/WORKDIR/04-18/45077",
27
+ "local_rank": 0,
28
+ "dataset": "dataset",
29
+ "config_file": "models/week0417/json/single_scale_grouping_ft.json",
30
+ "lambda_L1": 1,
31
+ "lambda_Perceptual": 1.0,
32
+ "lambda_PatchGAN": 1.0,
33
+ "lambda_GAN": 1,
34
+ "add_gan_epoch": 0,
35
+ "lambda_kld_loss": 1e-06,
36
+ "lambda_style_loss": 1.0,
37
+ "lambda_feat": 10.0,
38
+ "use_slic": true,
39
+ "patch_size": 64,
40
+ "netPatchD_scale_capacity": 4.0,
41
+ "netPatchD_max_nc": 384,
42
+ "netPatchD": "StyleGAN2",
43
+ "use_antialias": true,
44
+ "patch_use_aggregation": false,
45
+ "lambda_R1": 1.0,
46
+ "lambda_ffl_loss": 1.0,
47
+ "lambda_patch_R1": 1.0,
48
+ "R1_once_every": 16,
49
+ "add_self_loops": 1,
50
+ "test_time": false,
51
+ "sp_num": 196,
52
+ "label_path": "/home/xtli/DATA/BSR/BSDS500/data/groundTruth",
53
+ "model_name": "model",
54
+ "num_D": 2,
55
+ "n_layers_D": 3,
56
+ "n_cluster": 10,
57
+ "temperature": 23,
58
+ "add_gcn_epoch": 0,
59
+ "add_clustering_epoch": 0,
60
+ "add_texture_epoch": 0,
61
+ "dec_input_mode": "sine_wave_noise",
62
+ "sine_weight": true
63
+ }