From c22c4e803c7f824f7913359aa10b7967dff9a5ce Mon Sep 17 00:00:00 2001 From: gachiemchiep Date: Tue, 3 Mar 2020 11:23:50 +0900 Subject: [PATCH] Fix bug: RuntimeError: set_sizes_contiguous is not allowed on Tensor created from .data or .detach() --- test_net.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/test_net.py b/test_net.py index 2377aeb..edf6b70 100644 --- a/test_net.py +++ b/test_net.py @@ -272,11 +272,11 @@ def parse_args(): else: for i,index in enumerate(ratio_index_vu[0]): data = next(data_iter_vu) - im_data.data.resize_(data[0].size()).copy_(data[0]) - query.data.resize_(data[1].size()).copy_(data[1]) - im_info.data.resize_(data[2].size()).copy_(data[2]) - gt_boxes.data.resize_(data[3].size()).copy_(data[3]) - catgory.data.resize_(data[4].size()).copy_(data[4]) + im_data.resize_(data[0].size()).copy_(data[0]) + query.resize_(data[1].size()).copy_(data[1]) + im_info.resize_(data[2].size()).copy_(data[2]) + gt_boxes.resize_(data[3].size()).copy_(data[3]) + catgory.resize_(data[4].size()).copy_(data[4]) # Run Testing