Skip to content

Commit 2981245

Browse files
authored
Update test.py (open-mmlab#93)
1 parent bafc0e5 commit 2981245

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mmseg/apis/test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def single_gpu_test(model, data_loader, show=False, out_dir=None):
3131
for i, data in enumerate(data_loader):
3232
with torch.no_grad():
3333
result = model(return_loss=False, **data)
34-
if isinstance(results, list):
34+
if isinstance(result, list):
3535
results.extend(result)
3636
else:
3737
results.append(result)
@@ -96,7 +96,7 @@ def multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=False):
9696
for i, data in enumerate(data_loader):
9797
with torch.no_grad():
9898
result = model(return_loss=False, rescale=True, **data)
99-
if isinstance(results, list):
99+
if isinstance(result, list):
100100
results.extend(result)
101101
else:
102102
results.append(result)

0 commit comments

Comments
 (0)