predict batch returns wrong results when given a batch of images
predict batch returns wrong results when given a batch of images: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). predict batch returns wrong results when given a batch of images I'm trying to speed up inference by running prediction on a stack of images at once instead of…
The task
predict_batch returns wrong results when given a batch of images I'm trying to speed up inference by running prediction on a stack of images at once instead of looping over them in Python. I have a bunch of pre-processed image tensors stacked into shape (B, C, H, W) and I'm…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.