From 429c8b3fbf5fdd70d4edb31a1461756c88f503da Mon Sep 17 00:00:00 2001 From: MY_ Date: Tue, 25 Aug 2020 21:12:39 +0800 Subject: [PATCH] Update test.py (#93) --- mmseg/apis/test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py index 5b4b2da40..7f98abf29 100644 --- a/mmseg/apis/test.py +++ b/mmseg/apis/test.py @@ -31,7 +31,7 @@ def single_gpu_test(model, data_loader, show=False, out_dir=None): for i, data in enumerate(data_loader): with torch.no_grad(): result = model(return_loss=False, **data) - if isinstance(results, list): + if isinstance(result, list): results.extend(result) else: results.append(result) @@ -96,7 +96,7 @@ def multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=False): for i, data in enumerate(data_loader): with torch.no_grad(): result = model(return_loss=False, rescale=True, **data) - if isinstance(results, list): + if isinstance(result, list): results.extend(result) else: results.append(result)