diff --git a/direct/nn/crossdomain/crossdomain.py b/direct/nn/crossdomain/crossdomain.py index 3123cc9f..39fc4259 100644 --- 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 index 8a54bea6..ab985e40 100644 --- 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 index 9760be80..b41cb460 100644 --- 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 index 507ff173..5a63722a 100644 --- 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",