Skip to content

Commit 758a946

Browse files
author
Aleksei Silvestrov
committed
3 celeba model configs
1 parent 61dce17 commit 758a946

6 files changed

+226
-2
lines changed

‎README.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@ Docker: TODO
191191
On the host machine:
192192
# Make shure you are in lama folder
193193
cd lama
194+
export TORCH_HOME=$(pwd) && export PYTHONPATH=.
194195
195196
# Download CelebA-HQ dataset
196197
# Download data256x256.zip from https://drive.google.com/drive/folders/11Vz0fqHS2rXDb5pprgTjpD7S2BAJhi1P

‎configs/training/data/abl-04-256-mh-dist-celeba.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# @package _group_
22

3-
batch_size: 10
3+
batch_size: 5
44
val_batch_size: 3
55
num_workers: 3
66

‎configs/training/e.logacheva_2021-10-13_01-54-53_train_ablv2_work_no_segmpl_csirpl_celeba_csirpl03_new.yaml‎

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
run_title: ''
2+
3+
visualizer:
4+
kind: directory
5+
outdir: ./visualizer-output/celeba/train_ablv2_work_no_segmpl_csirpl_celeba_csirpl03_new/samples
6+
key_order:
7+
- image
8+
- predicted_image
9+
- discr_output_fake
10+
- discr_output_real
11+
- inpainted
12+
rescale_keys:
13+
- discr_output_fake
14+
- discr_output_real
15+
trainer:
16+
kwargs:
17+
gpus: -1
18+
accelerator: ddp
19+
max_epochs: 40
20+
gradient_clip_val: 1
21+
log_gpu_memory: None
22+
limit_train_batches: 25000
23+
val_check_interval: 2600
24+
log_every_n_steps: 250
25+
precision: 32
26+
terminate_on_nan: false
27+
check_val_every_n_epoch: 1
28+
num_sanity_val_steps: 8
29+
replace_sampler_ddp: false
30+
checkpoint_kwargs:
31+
verbose: true
32+
save_top_k: 5
33+
save_last: true
34+
period: 1
35+
monitor: val_ssim_fid100_f1_total_mean
36+
mode: max
37+
training_model:
38+
kind: default
39+
visualize_each_iters: 1000
40+
concat_mask: true
41+
store_discr_outputs_for_vis: true
42+
losses:
43+
l1:
44+
weight_missing: 0
45+
weight_known: 10
46+
perceptual:
47+
weight: 0
48+
adversarial:
49+
kind: r1
50+
weight: 10
51+
gp_coef: 0.001
52+
mask_as_fake_target: true
53+
allow_scale_mask: true
54+
feature_matching:
55+
weight: 100
56+
segm_pl:
57+
weight: 0.3
58+
arch_encoder: resnet50
59+
imagenet_weights: true
60+
optimizers:
61+
generator:
62+
kind: adam
63+
lr: 0.001
64+
discriminator:
65+
kind: adam
66+
lr: 0.0001
67+
68+
defaults:
69+
- location: celeba
70+
- data: abl-04-256-mh-dist-celeba
71+
- evaluator: default_inpainted
72+
- generator: pix2pixhd_global_sigmoid
73+
- discriminator: pix2pixhd_nlayer
74+
- hydra: overrides

‎configs/training/e.logacheva_2021-10-13_02-00-48_train_ablv2_work_no_segmpl_vgg_celeba_l2_vgg003_new.yaml‎

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
run_title: ''
2+
3+
visualizer:
4+
kind: directory
5+
outdir: ./visualizer-output/celeba/train_ablv2_work_no_segmpl_vgg_celeba_l2_vgg003_new/samples
6+
key_order:
7+
- image
8+
- predicted_image
9+
- discr_output_fake
10+
- discr_output_real
11+
- inpainted
12+
rescale_keys:
13+
- discr_output_fake
14+
- discr_output_real
15+
trainer:
16+
kwargs:
17+
gpus: -1
18+
accelerator: ddp
19+
max_epochs: 40
20+
gradient_clip_val: 1
21+
log_gpu_memory: None
22+
limit_train_batches: 25000
23+
val_check_interval: 2600
24+
log_every_n_steps: 250
25+
precision: 32
26+
terminate_on_nan: false
27+
check_val_every_n_epoch: 1
28+
num_sanity_val_steps: 8
29+
replace_sampler_ddp: false
30+
checkpoint_kwargs:
31+
verbose: true
32+
save_top_k: 5
33+
save_last: true
34+
period: 1
35+
monitor: val_ssim_fid100_f1_total_mean
36+
mode: max
37+
training_model:
38+
kind: default
39+
visualize_each_iters: 1000
40+
concat_mask: true
41+
store_discr_outputs_for_vis: true
42+
losses:
43+
l1:
44+
weight_missing: 0
45+
weight_known: 10
46+
perceptual:
47+
weight: 0.03
48+
kwargs:
49+
metric: l2
50+
adversarial:
51+
kind: r1
52+
weight: 10
53+
gp_coef: 0.001
54+
mask_as_fake_target: true
55+
allow_scale_mask: true
56+
feature_matching:
57+
weight: 100
58+
segm_pl:
59+
weight: 0
60+
optimizers:
61+
generator:
62+
kind: adam
63+
lr: 0.001
64+
discriminator:
65+
kind: adam
66+
lr: 0.0001
67+
68+
defaults:
69+
- location: celeba
70+
- data: abl-04-256-mh-dist-celeba
71+
- evaluator: default_inpainted
72+
- generator: pix2pixhd_global_sigmoid
73+
- discriminator: pix2pixhd_nlayer
74+
- hydra: overrides

‎configs/training/e.logacheva_2021-10-13_02-10-33_train_ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new.yaml‎

Lines changed: 75 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,75 @@
1+
run_title: ''
2+
3+
visualizer:
4+
kind: directory
5+
outdir: ./visualizer-output/celeba/train_ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new/samples
6+
key_order:
7+
- image
8+
- predicted_image
9+
- discr_output_fake
10+
- discr_output_real
11+
- inpainted
12+
rescale_keys:
13+
- discr_output_fake
14+
- discr_output_real
15+
trainer:
16+
kwargs:
17+
gpus: -1
18+
accelerator: ddp
19+
max_epochs: 40
20+
gradient_clip_val: 1
21+
log_gpu_memory: None
22+
limit_train_batches: 25000
23+
val_check_interval: 2600
24+
log_every_n_steps: 250
25+
precision: 32
26+
terminate_on_nan: false
27+
check_val_every_n_epoch: 1
28+
num_sanity_val_steps: 8
29+
replace_sampler_ddp: false
30+
benchmark: true
31+
resume_from_checkpoint: /group-volume/User-Driven-Content-Generation/e.logacheva/CelebA-HQ-inpainting/experiments/e.logacheva_2021-10-12_21-37-20_train_ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new/models/last.ckpt
32+
checkpoint_kwargs:
33+
verbose: true
34+
save_top_k: 5
35+
save_last: true
36+
period: 1
37+
monitor: val_ssim_fid100_f1_total_mean
38+
mode: max
39+
training_model:
40+
kind: default
41+
visualize_each_iters: 1000
42+
concat_mask: true
43+
store_discr_outputs_for_vis: true
44+
losses:
45+
l1:
46+
weight_missing: 0
47+
weight_known: 10
48+
perceptual:
49+
weight: 0
50+
adversarial:
51+
kind: r1
52+
weight: 10
53+
gp_coef: 0.001
54+
mask_as_fake_target: true
55+
allow_scale_mask: true
56+
feature_matching:
57+
weight: 100
58+
segm_pl:
59+
weight: 1
60+
imagenet_weights: true
61+
optimizers:
62+
generator:
63+
kind: adam
64+
lr: 0.001
65+
discriminator:
66+
kind: adam
67+
lr: 0.0001
68+
69+
defaults:
70+
- location: celeba
71+
- data: abl-04-256-mh-dist-celeba
72+
- evaluator: default_inpainted
73+
- generator: pix2pixhd_global_sigmoid
74+
- discriminator: pix2pixhd_nlayer
75+
- hydra: overrides

‎configs/training/lama-regular.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,4 +61,4 @@ defaults:
6161
- data: abl-04-256-mh-dist
6262
- evaluator: default_inpainted
6363
- trainer: any_gpu_large_ssim_ddp_final
64-
- hydra: overrides
64+
- hydra: overrides

0 commit comments

Comments
 (0)