Как обучить модель на всем наборе данных?

history1=model.fit(X_train,y_trainHot,batch_size=128,epochs=30,validation_split=0.1, shuffle=True)

выдает ошибку

InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run GatherV2: Dst tensor is not initialized. [Op:GatherV2]

однако, если я уменьшу выборку, например

history1=model.fit(X_train[:4000],y_trainHot[:4000],batch_size=128,epochs=30,validation_split=0.1, shuffle=True)

то модель нормально начинает обучаться Как можно исправить эту ошибку ???