[ERROR]: mask_trunc array size mismatch on custom dataset

[Lokesh-26]

I am trying to train my own dataset. The array size does not match when I run the following command ./core/gdrn_modeling/train_gdrn.sh configs/gdrn/br6d/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_br6d.py 0 I am not sure reshaping the array would make it better. Can you please guide me on resolving this error?

ValueError: Caught ValueError in DataLoader worker process 0. Original Traceback (most recent call last): File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/.venv/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/.venv/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/.venv/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/core/gdrn_modeling/../../core/gdrn_modeling/datasets/data_loader_online.py", line 764, in __getitem__ processed_data = self.read_data(dataset_dict) File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/core/gdrn_modeling/../../core/base_data_loader.py", line 93, in read_data return self.read_data_train(dataset_dict) File "/media/gouda/3C448DDD448D99F2/segmentation/gdrnpp_bop2022/core/gdrn_modeling/../../core/gdrn_modeling/datasets/data_loader_online.py", line 520, in read_data_train mask_trunc = mask_visib * mask_trunc.astype("float32") ValueError: operands could not be broadcast together with shapes (395,640) (1200,1944)