【寒武纪硬件产品型号】必填*：MLU370

【使用操作系统】必填*：ubuntu20.04

【使用驱动版本】必填*：v4.20.18

【出错信息】必填*：我使用的是Pytorch-Magicmind。使用Image.open加载的数据集如何增大batch_size？我的推理代码不是DataLoader。代码贴于github。如果想进一步优化推理性能，使用这种数据集加载方式还有什么措施吗？

【出错代码链接】https://github.com/LSJ5106/MLU370inference/blob/main/val_SL_mlu3.py