2021-09-08

当出现这种错误，且经检查发现输入数据的维度和模型相互匹配时，记得查看任务管理器中的GPU负荷。有较大可能为GPU显存超载导致。
此时需添加代码：
physical_devices = tf.config.list_physical_devices(‘GPU’)
tf.config.experimental.set_memory_growth(physical_devices[0], enable=True)
避免被tensorflow初始锁定过多显存。