Skip to content

GitLab

"tests/ocr_ppocr_server_params.txt" did not exist on "b4d7be33f0a37bf729316491cc75e927895d5bff"
Unverified Commit 9de346b2 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Update train.py

parent b8a65d43
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tools/train.py
View file @ 9de346b2
...@@ -97,8 +97,7 @@ def main(config, device, logger, vdl_writer): ...@@ -97,8 +97,7 @@ def main(config, device, logger, vdl_writer):
# build metric # build metric
eval_class = build_metric(config['Metric']) eval_class = build_metric(config['Metric'])
# load pretrain model # load pretrain model
#pre_best_model_dict = load_dygraph_params(config, model, logger, optimizer) pre_best_model_dict = load_dygraph_params(config, model, logger, optimizer)
pre_best_model_dict = {}
logger.info('train dataloader has {} iters'.format(len(train_dataloader))) logger.info('train dataloader has {} iters'.format(len(train_dataloader)))
if valid_dataloader is not None: if valid_dataloader is not None:
logger.info('valid dataloader has {} iters'.format( logger.info('valid dataloader has {} iters'.format(
......
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