RuntimeError: The size of tensor a (0) must match the size of tensor b (5) at non-singleton dimension 2

[huanxi-2018]

Hello， everyone，i have this question,could you help me?

Traceback (most recent call last):
File "./tools/train.py", line 240, in
main()
File "./tools/train.py", line 236, in main
train_model(model,datasets,cfg,distributed=distributed,validate=(not args.no_validate),timestamp=timestamp,meta=meta,
File "/home3/HWGroup/wangcy/JunLyu/zxk/mmediting/mmedit/apis/train.py", line 109, in train_model
_non_dist_train(#这里出错
File "/home3/HWGroup/wangcy/JunLyu/zxk/mmediting/mmedit/apis/train.py", line 375, in _non_dist_train
runner.run(data_loaders, cfg.workflow, cfg.total_iters)
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/mmcv/runner/iter_based_runner.py", line 144, in run
iter_runner(iter_loaders[i], **kwargs)
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/mmcv/runner/iter_based_runner.py", line 70, in train
self.call_hook('after_train_iter')
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/mmcv/runner/base_runner.py", line 317, in call_hook
getattr(hook, fn_name)(self)
File "/home3/HWGroup/wangcy/JunLyu/zxk/mmediting/mmedit/core/evaluation/eval_hooks.py", line 42, in after_train_iter
results = single_gpu_test(
File "/home3/HWGroup/wangcy/JunLyu/zxk/mmediting/mmedit/apis/test.py", line 43, in single_gpu_test
result = model(
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/mmcv/parallel/data_parallel.py", line 51, in forward
return super().forward(*inputs, **kwargs)
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 166, in forward
return self.module(*inputs[0], **kwargs[0])
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 116, in new_func
return old_func(*args, **kwargs)
File "/home3/HWGroup/wangcy/JunLyu/zxk/mmediting/mmedit/models/restorers/basic_restorer.py", line 75, in forward
return self.forward_test(lq, gt, **kwargs)
File "/home3/HWGroup/wangcy/JunLyu/zxk/DAVSR-1.0.0/davsr/models/restorers/davsr_model.py", line 181, in forward_test
output_i = self.generator(lq_clip).cpu()
File "/home3/HWGroup/wangcy/.conda/envs/davsr-pytorch/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in call_impl
return forward_call(*input, **kwargs)
File "/home3/HWGroup/wangcy/JunLyu/zxk/DAVSR-1.0.0/davsr/models/backbones/sr_backbones/davsr_net.py", line 988, in forward
FB = ps2ot(k, (sf[0]*t, sf[1]*w, sf[2]*h))#FB是光学传递函数的输出 # [1, 1, 100, 256, 256]
File "/home3/HWGroup/wangcy/JunLyu/zxk/DAVSR-1.0.0/davsr/models/backbones/sr_backbones/davsr_net.py", line 549, in ps2ot
otf[:, :, :psf.shape[2], :psf.shape[3], :psf.shape[4]].copy(psf) # [1, 1, 100, 256, 256]
RuntimeError: The size of tensor a (0) must match the size of tensor b (5) at non-singleton dimension 2