Skip to content

GitLab

"...composable_kernel.git" did not exist on "05e0467a1a7ea705e75356aab58072007de53d6f"
Commit d3dcaed4 authored by weishengyu's avatar weishengyu
Browse files

change config

parent f2d98c5e
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
tools/style_text_rec/configs/config.yml
View file @ d3dcaed4
...@@ -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
......
tools/style_text_rec/configs/dataset_config.yml
View file @ d3dcaed4
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment