Allocator (GPU_0_bfc) ran out of memory trying to allocate 128.00MiB

各位大佬，我在训练一个模型，基于Keras，用的是5折交叉验证，然后前3折都没问题，但是在进行第4折时就会提示Allocator (GPU_0_bfc) ran out of memory trying to allocate 128.00MiB.这应该是gpu内存不足了吧。也试过在每折验证前加 K.clear_session() tf.reset_default_graph()，但是程序就报错了： alueError: Tensor(“conv3d_1_1/concat:0”, shape=(?, 256, 256, 8, 1), dtype=float32, device=/device:CPU:0) must be from the same graph as Tensor(“loss/conv3d_1_loss/ones:0”, shape=(?, ?, ?, ?, ?), dtype=float32)，请问这到底是是啥原因呀？

网络比较复杂，batch size比较小，然后输入2562568，6块 TITAN Xp 12G，求助~~~

-------------------------------------分割线-------------------------------
问题已经得到解决，哈哈，牵扯到一个模型的load和save问题。。。噗