Spaces:
Runtime error
Runtime error
sunshineatnoon
commited on
Commit
·
2bff7bc
1
Parent(s):
2e1db3f
Add application file
Browse files- weights/.gitattributes +1 -0
- weights/108073/cpk.pth +3 -0
- weights/108073/exp_args.json +63 -0
- weights/108073/exp_args.txt +60 -0
- weights/12003/cpk.pth +3 -0
- weights/12003/exp_args.json +63 -0
- weights/12003/exp_args.txt +60 -0
- weights/12074/cpk.pth +3 -0
- weights/12074/exp_args.json +63 -0
- weights/12074/exp_args.txt +60 -0
- weights/134008/cpk.pth +3 -0
- weights/134008/exp_args.json +63 -0
- weights/134052/cpk.pth +3 -0
- weights/134052/exp_args.json +63 -0
- weights/138032/cpk.pth +3 -0
- weights/138032/exp_args.json +63 -0
- weights/145053/cpk.pth +3 -0
- weights/145053/exp_args.json +63 -0
- weights/164074/cpk.pth +3 -0
- weights/164074/exp_args.json +63 -0
- weights/169012/cpk.pth +3 -0
- weights/169012/exp_args.json +63 -0
- weights/169012/exp_args.txt +60 -0
- weights/198023/cpk.pth +3 -0
- weights/198023/exp_args.json +63 -0
- weights/198023/exp_args.txt +60 -0
- weights/25098/cpk.pth +3 -0
- weights/25098/exp_args.json +63 -0
- weights/25098/exp_args.txt +60 -0
- weights/277095/cpk.pth +3 -0
- weights/277095/exp_args.json +63 -0
- weights/45077/cpk.pth +3 -0
- weights/45077/exp_args.json +63 -0
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 |
+
}
|