Merge pull request #521 from mwawrzos/torchhub

[SSD][hubconf] contiguous input tensor
2020-05-26 19:31:18 +02:00 · 2020-05-26 19:31:18 +02:00 · 3e961c3a14
parent 9189f07093 ffe84ae70a
commit 3e961c3a14
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: