Update train.py

This commit is contained in:
Double_V 2021-07-09 09:27:07 +08:00 committed by GitHub
parent b8a65d4333
commit 9de346b234
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions

View File

@ -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(