VISTA icon indicating copy to clipboard operation
VISTA copied to clipboard

Can the input batchsize be more than 1?

Open flatblackwhite opened this issue 2 years ago • 0 comments

when setting batchsize more than 1,will report an error“RuntimeError: stack expects each tensor to be equal size, but got [1, 185, 128, 101] at entry 0 and [1, 249, 188, 213] at entry 1 Collate error on the key 'image' of dictionary data.“ I only use TotalSegmentator to train.

flatblackwhite avatar Aug 11 '23 03:08 flatblackwhite