Невозможно обучить модель, используя Ret iNet, используя CNN

Question

Я пытался построить Ret iNet по документу https://ieeexplore.ieee.org/document/8561051. Ниже приведен мой код `

model = models.Sequential()
model.add(layers.Conv2D(64, (3, 3), strides=(1,1), kernel_initializer='he_uniform', input_shape=(512, 512,3) ))
model.add(layers.MaxPooling2D((3, 3), strides=(1, 1)))
model.add(layers.Conv2D(192, (3, 3),strides=(1,1)))
model.add(layers.MaxPooling2D((3, 3), strides=(1, 1)))
model.add(layers.LeakyReLU())
model.add(layers.Conv2D(384, (3, 3),strides=(1,1)))
model.add(layers.Conv2D(256, (3, 3),strides=(1,1)))
model.add(layers.Conv2D(256, (3, 3), strides=(1,1)))
model.add(layers.MaxPooling2D((3, 3), strides=(1, 1)))
model.add(layers.LeakyReLU())
model.add(layers.Flatten())
model.add(layers.Dense(1,activity_regularizer=tf.keras.regularizers.l2(l=0.0005)))
model.summary()
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
model.fit(X_train , Y_train , epochs = 30, batch_size = 16)
` 

Но на тренировке это показало ошибку: ResourceExhaustedError введите описание изображения здесь . Пожалуйста, помогите решить эту проблему. X_train, Y_train имеет форму ((558, 512, 512, 3), (558, 1))