"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "c9799991f2e52cd3b2e331c52114007608244807"
Commit ef0880b9 authored by WenmuZhou's avatar WenmuZhou
Browse files

delete save_inference_mode fun, because the dev paddle has support export crnn model

parent 606a3873
...@@ -333,22 +333,6 @@ def eval(model, valid_dataloader, post_process_class, eval_class): ...@@ -333,22 +333,6 @@ def eval(model, valid_dataloader, post_process_class, eval_class):
return metirc return metirc
def save_inference_mode(model, config, logger):
if dist.get_rank() == 0:
model.eval()
print('infer')
save_path = '{}/infer/{}'.format(config['Global']['save_model_dir'],
config['Architecture']['model_type'])
if config['Architecture']['model_type'] == 'rec':
input_shape = [None, 3, 32, None]
jit_model = paddle.jit.to_static(
model, input_spec=[paddle.static.InputSpec(input_shape)])
paddle.jit.save(jit_model, save_path)
logger.info('inference model save to {}'.format(save_path))
model.train()
def preprocess(): def preprocess():
FLAGS = ArgsParser().parse_args() FLAGS = ArgsParser().parse_args()
config = load_config(FLAGS.config) config = load_config(FLAGS.config)
......
...@@ -89,7 +89,6 @@ def main(config, device, logger, vdl_writer): ...@@ -89,7 +89,6 @@ def main(config, device, logger, vdl_writer):
program.train(config, train_dataloader, valid_dataloader, device, model, program.train(config, train_dataloader, valid_dataloader, device, model,
loss_class, optimizer, lr_scheduler, post_process_class, loss_class, optimizer, lr_scheduler, post_process_class,
eval_class, pre_best_model_dict, logger, vdl_writer) eval_class, pre_best_model_dict, logger, vdl_writer)
program.save_inference_mode(model, config, logger)
def test_reader(config, device, logger): def test_reader(config, device, logger):
......
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