Fail to train my own data

[tymanab]

Hi, Thanks for sharing this great project. I want to try run this project with my own dataset, but the training phased failed when calculating object cross entropy loss. The images are prepared follow Readme.

Here is the error message I got:

Exception has occurred: RuntimeError input and target batch or spatial sizes don't match: target [1, 952, 535], input [1, 256, 952, 534] File "/home/gw/projects/gaussian-grouping/train.py", line 89, in training loss_obj = cls_criterion(logits.unsqueeze(0), gt_obj.unsqueeze(0)).squeeze().mean() File "/home/gw/projects/gaussian-grouping/train.py", line 255, in training(lp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, args.use_wandb) RuntimeError: input and target batch or spatial sizes don't match: target [1, 952, 535], input [1, 256, 952, 534]

I also tried some other datasets, and most of them share similar problems. Lerf mask all works well, but mipnerf360 data(from the original website) such as room, garden will also get stuck here. Is there any extra preparation I need to do with my own dataset?