diff --git a/upsnet/upsnet_end2end_test.py b/upsnet/upsnet_end2end_test.py
index cd8c8d6..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]
 
@@ -231,9 +229,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