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

polish for mixup

Browse Source
This commit is contained in:
WuHaobo 2020-04-27 21:33:25 +08:00
parent 38728f638b
commit 165068cea8
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/program.py
View File

@ -398,7 +398,6 @@ def run(dataloader, exe, program, fetchs, epoch=0, mode='train'):
else:
logger.info("END epoch:{:<3d} {:s} {:s}s".format(epoch, mode, end_str))
# save the best model
# return top1_acc in order to save the best model
if mode == 'valid':
top1_acc = fetchs["top1"][1].avg
return top1_acc
return fetchs["top1"][1].avg