mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

use compiled program

pull/76/head
WuHaobo 2020-04-26 09:35:49 +08:00
parent 9acc8a7544
commit c02e1fb197
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/eval.py
View File

@ -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')

5
tools/train.py
View File

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