Question about training process.

[TMDTom]

Nice work and I'm trying to use this code to run on my data. But this error I get now. reate web directory ./checkpoints/hieu_mr_ct_pelvis_attgan/web... Traceback (most recent call last): File "/tmp/code/MaskGan/train.py", line 98, in model.optimize_parameters() # calculate loss functions, get gradients, update network weights File "/tmp/code/MaskGan/models/mask_gan_model.py", line 231, in optimize_parameters self.forward() # compute fake images and reconstruction images. File "/tmp/code/MaskGan/models/mask_gan_model.py", line 139, in forward self.fake_A, self.outputs_A, self.attentions_A, self.images_A = self.netG_B(self.real_B) # G_B(B) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, *kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/parallel/data_parallel.py", line 183, in forward return self.module(inputs[0], module_kwargs[0]) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, *kwargs) File "/tmp/code/MaskGan/models/networks.py", line 438, in forward x = F.relu(self.conv1_norm(self.conv1(x))) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(args, kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/conv.py", line 460, in forward return self._conv_forward(input, self.weight, self.bias) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/conv.py", line 456, in _conv_forward return F.conv2d(input, weight, bias, self.stride, File "/usr/local/lib/python3.10/site-packages/apex/amp/wrap.py", line 21, in wrapper args[i] = utils.cached_cast(cast_fn, args[i], handle.cache) File "/usr/local/lib/python3.10/site-packages/apex/amp/utils.py", line 97, in cached_cast if cached_x.grad_fn.next_functions[1][0].variable is not x: IndexError: tuple index out of range

Ask for help, thanks a lot