From 9de346b2347fcc67c3219119be5980824878835b Mon Sep 17 00:00:00 2001 From: Double_V Date: Fri, 9 Jul 2021 09:27:07 +0800 Subject: [PATCH] Update train.py --- tools/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/train.py b/tools/train.py index 2091ff48..05d295aa 100755 --- a/tools/train.py +++ b/tools/train.py @@ -97,8 +97,7 @@ def main(config, device, logger, vdl_writer): # build metric eval_class = build_metric(config['Metric']) # load pretrain model - #pre_best_model_dict = load_dygraph_params(config, model, logger, optimizer) - pre_best_model_dict = {} + pre_best_model_dict = load_dygraph_params(config, model, logger, optimizer) logger.info('train dataloader has {} iters'.format(len(train_dataloader))) if valid_dataloader is not None: logger.info('valid dataloader has {} iters'.format( -- GitLab