提交 d3dcaed4 编写于 作者: W weishengyu

change config

上级 f2d98c5e
...@@ -23,7 +23,7 @@ Predictor: ...@@ -23,7 +23,7 @@ Predictor:
- 0.5 - 0.5
expand_result: false expand_result: false
bg_generator: bg_generator:
pretrain: style_text_models/bg_generator pretrain: models/style_text_rec/bg_generator
module_name: bg_generator module_name: bg_generator
generator_type: BgGeneratorWithMask generator_type: BgGeneratorWithMask
encode_dim: 64 encode_dim: 64
...@@ -33,7 +33,7 @@ Predictor: ...@@ -33,7 +33,7 @@ Predictor:
conv_block_dilation: true conv_block_dilation: true
output_factor: 1.05 output_factor: 1.05
text_generator: text_generator:
pretrain: style_text_models/text_generator pretrain: models/style_text_rec/text_generator
module_name: text_generator module_name: text_generator
generator_type: TextGenerator generator_type: TextGenerator
encode_dim: 64 encode_dim: 64
...@@ -42,7 +42,7 @@ Predictor: ...@@ -42,7 +42,7 @@ Predictor:
conv_block_dropout: false conv_block_dropout: false
conv_block_dilation: true conv_block_dilation: true
fusion_generator: fusion_generator:
pretrain: style_text_models/fusion_generator pretrain: models/style_text_rec/fusion_generator
module_name: fusion_generator module_name: fusion_generator
generator_type: FusionGeneratorSimple generator_type: FusionGeneratorSimple
encode_dim: 64 encode_dim: 64
......
...@@ -33,7 +33,7 @@ Predictor: ...@@ -33,7 +33,7 @@ Predictor:
- 0.5 - 0.5
expand_result: false expand_result: false
bg_generator: bg_generator:
pretrain: style_text_models/bg_generator pretrain: models/style_text_rec/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: ...@@ -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: style_text_models/text_generator pretrain: models/style_text_rec/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: ...@@ -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: style_text_models/fusion_generator pretrain: models/style_text_rec/fusion_generator
module_name: fusion_generator module_name: fusion_generator
generator_type: FusionGeneratorSimple generator_type: FusionGeneratorSimple
encode_dim: 64 encode_dim: 64
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册