tensorflow gpu2.4.0训练分类模型时在跑了几十个周期之后报错

Question

在用tf.kera训练cifar10数据时，出现如图所示的情况，我是用gpu训练模型的，但出错信息为什么说cpu分配出了问题，很多人说减少bitch-size，我尝试过了，结果都一样，反而增大bitch-size减少训练时间可以解决问题，但是以后训练集数据多了还是会碰到这种情况，所以想现在解决一下，有没有兄弟知道怎么解决的。在训练二十多个周期后，内存使用率高达99%，然后一直持续在99、98%这样，我的内存有16G，不知道这种情况是不是正常的。还有一个现象，它几乎每次都是训练到60或者61个周期才报错，这让我怀疑是不是和内存有关，毕竟如果是代码的问题，那应该一个周期都跑不起来。

代码如下