multiple images per gpu
Opened this issue · 1 comments
manongXT commented
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 commented