I use my own reference images, but encountering a problem as follow.
Traceback (most recent call last):
File "/home/champ/inference.py", line 321, in
main(cfg)
File "/home/champ/inference.py", line 299, in main
grid_video = torch.cat([ref_video_tensor, result_video_tensor], dim=0)
RuntimeError: Sizes of tensors must match except in dimension 0. Expected size 4 but got size 3 for tensor number 1 in the list.
I use my own reference images, but encountering a problem as follow. Traceback (most recent call last): File "/home/champ/inference.py", line 321, in
main(cfg)
File "/home/champ/inference.py", line 299, in main
grid_video = torch.cat([ref_video_tensor, result_video_tensor], dim=0)
RuntimeError: Sizes of tensors must match except in dimension 0. Expected size 4 but got size 3 for tensor number 1 in the list.