Learnable Cost Volume model

[danialzendehdel]

Can you help me with this?

Epoch: 0%| | 0/500 [00:17<?, ?it/s]
Traceback (most recent call last):
File "main.py", line 348, in
main()
File "main.py", line 286, in main
loss = train(imgU_crop, imgD_crop, disp_crop)
File "main.py", line 196, in train
output1, output2, output3 = model(imgU, imgD)
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 155, in forward
outputs = self.parallel_apply(replicas, inputs, kwargs)
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 165, in parallel_apply
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply
output.reraise()
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/_utils.py", line 395, in reraise
raise self.exc_type(msg)
TypeError: Caught TypeError in replica 0 on device 0.
Original Traceback (most recent call last):
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker
output = module(*input, **kwargs)
File "/ssoft/spack/arvine/v1/opt/spack/linux-rhel7-skylake_avx512/gcc-8.4.0/py-torch-1.6.0-43xbre3fdhzp6upz6mfe3jk6rpwt5uky/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in call_impl
result = self.forward(*input, **kwargs)
File "/work/vita/danial/360SD-Net/models/LCV_ours_sub3.py", line 187, in forward
refimg_fea.size()[3]).zero()).cuda()
TypeError: new(): argument 'size' must be tuple of ints, but found element of type float at pos 3