From c02e1fb197dce04534b9f59afb7bbea0eb33e383 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sun, 26 Apr 2020 09:35:49 +0800 Subject: [PATCH] use compiled program --- tools/eval.py | 5 ++--- tools/train.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/tools/eval.py b/tools/eval.py index ea0d8c309..db5ce4eec 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -71,9 +71,8 @@ def main(args): valid_reader = Reader(config, 'valid')() valid_dataloader.set_sample_list_generator(valid_reader, place) - - #compiled_valid_prog = program.compile(config, valid_prog) - compiled_valid_prog = valid_prog + + compiled_valid_prog = program.compile(config, valid_prog) program.run(valid_dataloader, exe, compiled_valid_prog, valid_fetchs, 0, 'valid') diff --git a/tools/train.py b/tools/train.py index effd6bc55..9cbe3edd8 100644 --- a/tools/train.py +++ b/tools/train.py @@ -24,7 +24,6 @@ from paddle.fluid.incubate.fleet.base import role_maker from paddle.fluid.incubate.fleet.collective import fleet from ppcls.data import Reader -from ppcls.utils import logger from ppcls.utils.config import get_config from ppcls.utils.save_load import init_model, save_model import program @@ -86,8 +85,8 @@ def main(args): if config.validate: valid_reader = Reader(config, 'valid')() valid_dataloader.set_sample_list_generator(valid_reader, place) - #compiled_valid_prog = program.compile(config, valid_prog) - compiled_valid_prog = valid_prog + compiled_valid_prog = program.compile(config, valid_prog) + compiled_train_prog = fleet.main_program for epoch_id in range(config.epochs): # 1. train with train dataset