Error during inference on custom video

[deepakkupanda]

====Note: Using block-wise evaluation with block size=None, overlap=None, batch size=None ====This may generate unwanted block artefacts. Loading the model... Using model STMFNet to upsample file birthday_girl_720x480p.yuv 0%| | 0/641 [00:00<?, ?it/s]/anaconda/envs/azureml_py38_PT_TF/lib/python3.8/site-packages/cupy/cuda/compiler.py:461: UserWarning: cupy.cuda.compile_with_cache has been deprecated in CuPy v10, and will be removed in the future. Use cupy.RawModule or cupy.RawKernel instead. warnings.warn( 0%| | 0/641 [00:00<?, ?it/s] Traceback (most recent call last): File "interpolate_yuv.py", line 136, in main() File "interpolate_yuv.py", line 118, in main out = model(frame0, frame1, frame2, frame3) File "/home/azureuser/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/deepakpanda7/code/Users/deepakpanda/slow-motion/ST-MFNet/models/stmfnet.py", line 287, in forward I1_us = self.upsampler(I1) File "/home/azureuser/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/deepakpanda7/code/Users/deepakpanda/slow-motion/ST-MFNet/models/misc/init.py", line 31, in forward im_up_row = F.conv2d(F.pad(im, pad=(p,p+1,0,0), mode='reflect'), self.filter, groups=3) RuntimeError: Expected object of scalar type Float but got scalar type Long for argument #2 'weight' in call to _thnn_conv_depthwise2d_forward