[SSD][hubconf] contiguous input tensor

2020-05-20 16:00:47 +02:00 · 2020-05-20 16:00:47 +02:00 · ffe84ae70a
parent 9189f07093
commit ffe84ae70a
1 changed files with 1 additions and 0 deletions
--- a/hubconf.py
+++ b/hubconf.py
@ -275,6 +275,7 @@ def nvidia_ssd_processing_utils():
            NHWC = np.array(inputs)
            NCHW = np.swapaxes(np.swapaxes(NHWC, 1, 3), 2, 3)
            tensor = torch.from_numpy(NCHW)
+            tensor = tensor.contiguous()
            tensor = tensor.cuda()
            tensor = tensor.float()
            if fp16: