Creating a model on the tensor slices

cifar_model_on_tensorslices.py

logs = "logs/"+datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
# Profiling on batches 200 to 220
tfboard_callback = tf.keras.callbacks.TensorBoard(
    log_dir=logs,
    histogram_freq=1,
    profile_batch='200,220'
)

history_dataset = model.fit(
    train_tensor_slices,
    validation_data=test_tensor_slices,
    epochs=EPOCHS,
    callbacks=[tfboard_callback]
)

%load_ext tensorboard
%tensorboard --logdir=logs