ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[2,256,72,72]

[hbx-mm]

前辈您好，看到您这样的良心代码很是激动，中文的解释和readme让我直接舍弃了别的代码，我尝试用自己数据集做一个训练，1000张图片，1个类别，我都按照你的readme做的，还担心显存不够（我的是win10环境，GTX1650，显存4G），将batchsize修改为2。可是还是出现了这个问题


不知道该怎么办了

[dodogoffy]

我觉得问题还是显存不够，把多尺度训练去掉，把batch_size调为1，把数据增强设为False再试试。