xintian-99 / oct2confocal_3dcyclegan Goto Github PK

I got the following error:

Traceback (most recent call last):
File "/kaggle/working/OCT2Confocal_3DCycleGAN/train.py", line 52, in
model.optimize_parameters() # calculate loss functions, get gradients, update network weights
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/cycle_gan_model.py", line 208, in optimize_parameters
self.backward_G() # calculate gradients for G_A and G_B
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/cycle_gan_model.py", line 193, in backward_G
self.loss_GL_A = networks.cal_gradient_loss_3d(self.real_A, self.fake_A, self.device) * lambda_GL
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/networks.py", line 327, in cal_gradient_loss_3d
grad_x_fake = F.conv3d(fake_data, filter_x, padding=1, groups=channels)
RuntimeError: expected stride to be a single integer value or a list of 4 values to match the convolution dimensions, but got stride=[1, 1, 1]

After this error I modify the line 237 in netwroks.py from
grad_x_fake = F.conv3d(fake_data, filter_x, padding=1, groups=channels)
to
grad_x_fake = F.conv3d(fake_data, filter_x, padding=1, groups=channels, stride=1)
and
grad_x_fake = F.conv3d(fake_data, filter_x, padding=1, groups=channels, stride=(1,1,1))
but the error remains.

Traceback (most recent call last):
File "/kaggle/working/OCT2Confocal_3DCycleGAN/train.py", line 52, in
model.optimize_parameters() # calculate loss functions, get gradients, update network weights
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/cycle_gan_model.py", line 208, in optimize_parameters
self.backward_G() # calculate gradients for G_A and G_B
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/cycle_gan_model.py", line 193, in backward_G
self.loss_GL_A = networks.cal_gradient_loss_3d(self.real_A, self.fake_A, self.device) * lambda_GL
File "/kaggle/working/OCT2Confocal_3DCycleGAN/models/networks.py", line 327, in cal_gradient_loss_3d
grad_x_fake = F.conv3d(fake_data, filter_x, padding=1, groups=channels, stride=1)
RuntimeError: expected stride to be a single integer value or a list of 4 values to match the convolution dimensions, but got stride=[1, 1, 1]

Does anyone have any suggestions?