[dnn/shift] many bugfixes in strided shift-conv

2019-07-10 19:49:31 -07:00
parent 4ca83f1935
commit 75cf2df110
4 changed files with 84 additions and 73 deletions
--- a/examples/python/pytorch/triton.py
+++ b/examples/python/pytorch/triton.py
@@ -152,7 +152,7 @@ class _ShiftConvNd(torch.nn.Module):

    def make_shift(self, kernel_size):
        if kernel_size == 3:
-            p = torch.Tensor([0., 1., 0.])
+            p = torch.Tensor([0.3, 0.4, 0.3])
        elif kernel_size == 5:
            p = torch.Tensor([0.1, 0.25, 0.3, 0.25, 0.1])
        elif kernel_size == 7: