jianzhangcs / ISTA-Net-PyTorch

ISTA-Net: Interpretable Optimization-Inspired Deep Network for Image Compressive Sensing, CVPR2018 (PyTorch Code)
228 stars 36 forks source link

Train-CS-MRI #8

Open xinyanzhang0032 opened 1 year ago

xinyanzhang0032 commented 1 year ago

您好 我在训练MRI_plus版本时,遇到维度不匹配的问题,具体的报错如下: x = x - self.lambda_step * fft_forback(x, mask) RuntimeError: The size of tensor a (2) must match the size of tensor b (256) at non-singleton dimension 5 请问这该如何解决呢?