Skip to content

Commit 259eeb0

Browse files
authored
fix efficient test bug (open-mmlab#702)
1 parent 813222a commit 259eeb0

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

mmseg/core/evaluation/eval_hooks.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,11 @@ def _do_evaluate(self, runner):
3131
return
3232

3333
from mmseg.apis import single_gpu_test
34-
results = single_gpu_test(runner.model, self.dataloader, show=False)
34+
results = single_gpu_test(
35+
runner.model,
36+
self.dataloader,
37+
show=False,
38+
efficient_test=self.efficient_test)
3539
runner.log_buffer.output['eval_iter_num'] = len(self.dataloader)
3640
key_score = self.evaluate(runner, results)
3741
if self.save_best:
@@ -84,7 +88,8 @@ def _do_evaluate(self, runner):
8488
runner.model,
8589
self.dataloader,
8690
tmpdir=tmpdir,
87-
gpu_collect=self.gpu_collect)
91+
gpu_collect=self.gpu_collect,
92+
efficient_test=self.efficient_test)
8893
if runner.rank == 0:
8994
print('\n')
9095
runner.log_buffer.output['eval_iter_num'] = len(self.dataloader)

0 commit comments

Comments
 (0)