multiple images per gpu

[manongXT]

I want to use " multiple images per gpu".in "# TODO: support multiple images per gpu (only minor changes are needed) dataset = build_dataset(cfg.data.test) data_loader = build_dataloader( dataset, imgs_per_gpu=2, workers_per_gpu=1, dist=distributed, shuffle=False)" I change the imgs_per_gpu=1 to imgs_per_gpu=2,but I got the error 'RuntimeError: stack expects each tensor to be equal size, but got [3, 128, 128] at entry 0 and [3, 128, 96] at entry 1' ,but I search the code ,i did not find the size of (128,96),so ,how can I use 'multiple images per gpu' to test.

[QinWinner]

I think you can check if you set ONNX_EXPORT=False in mmdet.deploy_params.py