|
|
@ -33,7 +33,7 @@ Predictor:
|
|
|
|
- 0.5
|
|
|
|
- 0.5
|
|
|
|
expand_result: false
|
|
|
|
expand_result: false
|
|
|
|
bg_generator:
|
|
|
|
bg_generator:
|
|
|
|
pretrain: models/style_text_rec/bg_generator
|
|
|
|
pretrain: style_text_models/bg_generator
|
|
|
|
module_name: bg_generator
|
|
|
|
module_name: bg_generator
|
|
|
|
generator_type: BgGeneratorWithMask
|
|
|
|
generator_type: BgGeneratorWithMask
|
|
|
|
encode_dim: 64
|
|
|
|
encode_dim: 64
|
|
|
@ -43,7 +43,7 @@ Predictor:
|
|
|
|
conv_block_dilation: true
|
|
|
|
conv_block_dilation: true
|
|
|
|
output_factor: 1.05
|
|
|
|
output_factor: 1.05
|
|
|
|
text_generator:
|
|
|
|
text_generator:
|
|
|
|
pretrain: models/style_text_rec/text_generator
|
|
|
|
pretrain: style_text_models/text_generator
|
|
|
|
module_name: text_generator
|
|
|
|
module_name: text_generator
|
|
|
|
generator_type: TextGenerator
|
|
|
|
generator_type: TextGenerator
|
|
|
|
encode_dim: 64
|
|
|
|
encode_dim: 64
|
|
|
@ -52,7 +52,7 @@ Predictor:
|
|
|
|
conv_block_dropout: false
|
|
|
|
conv_block_dropout: false
|
|
|
|
conv_block_dilation: true
|
|
|
|
conv_block_dilation: true
|
|
|
|
fusion_generator:
|
|
|
|
fusion_generator:
|
|
|
|
pretrain: models/style_text_rec/fusion_generator
|
|
|
|
pretrain: style_text_models/fusion_generator
|
|
|
|
module_name: fusion_generator
|
|
|
|
module_name: fusion_generator
|
|
|
|
generator_type: FusionGeneratorSimple
|
|
|
|
generator_type: FusionGeneratorSimple
|
|
|
|
encode_dim: 64
|
|
|
|
encode_dim: 64
|
|
|
|