issue: Make torch.where output contiguous before passing to fft.

NKI-AI · Jul 19, 2022 · cbce623 · cbce623
1 parent 6ded2eb
commit cbce623
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/direct/nn/crossdomain/crossdomain.py b/direct/nn/crossdomain/crossdomain.py
@@ -148,7 +148,7 @@ def _backward_operator(
                     sampling_mask == 0,
                     torch.tensor([0.0], dtype=kspace.dtype).to(kspace.device),
                     kspace,
-                ),
+                ).contiguous(),
                 self._spatial_dims,
             ),
             sensitivity_map,

diff --git a/direct/nn/kikinet/kikinet.py b/direct/nn/kikinet/kikinet.py
@@ -160,7 +160,7 @@ def forward(
                         sampling_mask == 0,
                         torch.tensor([0.0], dtype=kspace.dtype).to(kspace.device),
                         kspace,
-                    ),
+                    ).contiguous(),
                     self._spatial_dims,
                 ),
                 sensitivity_map,

diff --git a/direct/nn/lpd/lpd.py b/direct/nn/lpd/lpd.py
@@ -242,7 +242,7 @@ def _backward_operator(
                     sampling_mask == 0,
                     torch.tensor([0.0], dtype=kspace.dtype).to(kspace.device),
                     kspace,
-                ),
+                ).contiguous(),
                 self._spatial_dims,
             ),
             sensitivity_map,

diff --git a/setup.py b/setup.py
@@ -57,7 +57,7 @@ def finalize_options(self):
         "numpy>=1.21.2",
         "h5py==3.3.0",
         "omegaconf==2.1.1",
-        "torch==1.11.0",
+        "torch>=1.10.0",
         "torchvision",
         "scikit-image>=0.19.0",
         "scikit-learn>=1.0.1",