File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Expand file tree Collapse file tree 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):
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 )
You can’t perform that action at this time.
0 commit comments