mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #150 from littletomatodonkey/fix_fetch

Browse Source 
fix effnet train fetch
This commit is contained in:
littletomatodonkey 2020-06-08 18:24:02 +08:00 committed by GitHub
commit 38ad51ca0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/train.py
View File

@ -112,7 +112,7 @@ def main(args):
if config.validate and epoch_id % config.valid_interval == 0: if config.validate and epoch_id % config.valid_interval == 0:
if config.get('use_ema'): if config.get('use_ema'):
logger.info(logger.coloring("EMA validate start...")) logger.info(logger.coloring("EMA validate start..."))
with train_fetchs('ema').apply(exe): with ema.apply(exe):
top1_acc = program.run(valid_dataloader, exe, top1_acc = program.run(valid_dataloader, exe,
compiled_valid_prog, compiled_valid_prog,
valid_fetchs, epoch_id, 'valid') valid_fetchs, epoch_id, 'valid')