Setting up train/test split for use in Conv neural network

Question

I have a NN I\'d like to use here:

model = keras.models.Sequential([
    keras.layers.Conv2d(64, 7, activation="relu", padding="same",