a problrm in nodule_net.py

[huangguoqing1111]

File "D:\HGQ_DL\Nodule-CADx\src\net\nodule_net.py", line 420, in forward
self.mask_probs = data_parallel(self.mask_head, (torch.from_numpy(self.crop_boxes).cuda(), features))
File "D:\anaconda\envs\Stest\lib\site-packages\torch\nn\parallel\data_parallel.py", line 231, in data_parallel
return module(*inputs[0], **module_kwargs[0])
File "D:\anaconda\envs\Stest\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "D:\HGQ_DL\Nodule-CADx\src\net\nodule_net.py", line 236, in forward
up1 = f_4[b, :, z_start / 4:z_end / 4, y_start / 4:y_end / 4, x_start / 4:x_end / 4].unsqueeze(0)
TypeError: only integer tensors of a single element can be converted to an index

Hello , when I run this demo , I got this problem in the nodule_net.py , can you help me to solve it ? and the input is a .mhd file , it can be any file?

[huangguoqing1111]

@wenyalintw very thanks for help me

[huangguoqing1111]

I solve it , due to I use pytorc=1.13,so I need to change this code