OOM when allocating tensor with shape[5184,4,4,2048] and type float on /job:localhost/replica:0/task 报错原因:GPU内存不够,减小batch,或者减小模型。