File tree 1 file changed +7
-2
lines changed
1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -31,7 +31,11 @@ def _do_evaluate(self, runner):
31
31
return
32
32
33
33
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 )
35
39
runner .log_buffer .output ['eval_iter_num' ] = len (self .dataloader )
36
40
key_score = self .evaluate (runner , results )
37
41
if self .save_best :
@@ -84,7 +88,8 @@ def _do_evaluate(self, runner):
84
88
runner .model ,
85
89
self .dataloader ,
86
90
tmpdir = tmpdir ,
87
- gpu_collect = self .gpu_collect )
91
+ gpu_collect = self .gpu_collect ,
92
+ efficient_test = self .efficient_test )
88
93
if runner .rank == 0 :
89
94
print ('\n ' )
90
95
runner .log_buffer .output ['eval_iter_num' ] = len (self .dataloader )
You can’t perform that action at this time.
0 commit comments