"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "15a22fa05a63fad3374d0a0af0e0e4ec6e863add"
Unverified Commit 76bb40fc authored by Double_V's avatar Double_V Committed by GitHub
Browse files

fix ci

parent 54ed8d06
...@@ -55,6 +55,10 @@ def main(): ...@@ -55,6 +55,10 @@ def main():
model = build_model(config['Architecture']) model = build_model(config['Architecture'])
use_srn = config['Architecture']['algorithm'] == "SRN" use_srn = config['Architecture']['algorithm'] == "SRN"
if "model_type" in config['Architecture'].keys():
model_type = config['Architecture']['model_type']
else:
model_type = None
best_model_dict = init_model(config, model) best_model_dict = init_model(config, model)
if len(best_model_dict): if len(best_model_dict):
......
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