未验证 提交 38ad51ca 编写于 作者: L littletomatodonkey 提交者: GitHub

Merge pull request #150 from littletomatodonkey/fix_fetch

fix effnet train fetch
...@@ -112,7 +112,7 @@ def main(args): ...@@ -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')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册