RuntimeError: Expected 4-dimensional input for 4-dimensional weight [3, 1, 11, 11], but got 3-dimensional input of size [3, 900, 1600] instead

[yuxuanfanOrion]

Hi! Thanks for your great work.

I have met some bugs when I try to train bungee dataset.

The following are the bugs. Thanks for helping!

Traceback (most recent call last):
File "/home/fanyx2311/.conda/envs/sgs/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/fanyx2311/.conda/envs/sgs/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/main.py", line 39, in
cli.main()
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/server/cli.py", line 430, in main
run()
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/server/cli.py", line 284, in run_file
runpy.run_path(target, run_name="main")
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 321, in run_path
return _run_module_code(code, init_globals, run_name,
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 135, in _run_module_code
_run_code(code, mod_globals, init_globals,
File "/home/fanyx2311/.local/lib/python3.8/site-packages/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 124, in _run_code
exec(code, run_globals)
File "train.py", line 527, in
training(lp.extract(args), op.extract(args), pp.extract(args), dataset, args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, wandb, logger)
File "train.py", line 143, in training
ssim_loss = (1.0 - ssim(image, gt_image))
File "/data20_ssd/fanyx2311/Scaffold-GS/utils/loss_utils.py", line 41, in ssim
return _ssim(img1, img2, window, window_size, channel, size_average)
File "/data20_ssd/fanyx2311/Scaffold-GS/utils/loss_utils.py", line 44, in _ssim
mu1 = F.conv2d(img1, window, padding=window_size // 2, groups=channel)
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [3, 1, 11, 11], but got 3-dimensional input of size [3, 900, 1600] instead
Training progress: 0%|