From 21bad6e3cf2b7c3d1bef5d02aed4ddc56803d508 Mon Sep 17 00:00:00 2001 From: Evan Gebhardt Date: Fri, 14 Jun 2019 14:46:06 -0400 Subject: [PATCH 1/2] Update upsnet_end2end_test.py --- upsnet/upsnet_end2end_test.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/upsnet/upsnet_end2end_test.py b/upsnet/upsnet_end2end_test.py index cd8c8d6..292b88d 100644 --- a/upsnet/upsnet_end2end_test.py +++ b/upsnet/upsnet_end2end_test.py @@ -231,9 +231,7 @@ def upsnet_test(): for k, v in data.items(): data[k] = v.pin_memory().to(gpu_id, non_blocking=True) if torch.is_tensor(v) else v except StopIteration: - data = data.copy() - for k, v in data.items(): - data[k] = v.pin_memory().to(gpu_id, non_blocking=True) if torch.is_tensor(v) else v + break batch.append((data, None)) labels.append(label) i_iter += 1 From bd79f8d38f6a8c8b7c851573288af51dedeaaa81 Mon Sep 17 00:00:00 2001 From: Evan Gebhardt Date: Fri, 14 Jun 2019 14:54:46 -0400 Subject: [PATCH 2/2] Update upsnet_end2end_test.py --- upsnet/upsnet_end2end_test.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/upsnet/upsnet_end2end_test.py b/upsnet/upsnet_end2end_test.py index 292b88d..7e11955 100644 --- a/upsnet/upsnet_end2end_test.py +++ b/upsnet/upsnet_end2end_test.py @@ -62,8 +62,6 @@ def im_detect(output_all, data, im_infos): pred_pano_cls_inds_all = [] cls_inds_all = [] - if len(data) == 1: - output_all = [output_all] output_all = [{k: v.data.cpu().numpy() for k, v in output.items()} for output in output_all]