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?