mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Improve] replace loop of progressbar in api/test. (#878)

pull/1067/head
Hakjin Lee 2022-09-30 15:41:07 +09:00 committed by GitHub
parent 4eaaf89618
commit 1b4e9cd22a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmcls/apis/test.py
View File

@ -79,8 +79,7 @@ def single_gpu_test(model,
**show_kwargs) **show_kwargs)
batch_size = data['img'].size(0) batch_size = data['img'].size(0)
for _ in range(batch_size): prog_bar.update(batch_size)
prog_bar.update()
return results return results