CIFAR10 HPT Metrics

cifar10_hpt_metrics.py

import tensorflow as tf
import hypertune

hpt = hypertune.HyperTune()

class CustomCallback(tf.keras.callbacks.Callback):
  def on_epoch_end(self, epoch, logs=None):
    hpt.report_hyperparameter_tuning_metric(
      hyperparameter_metric_tag='val_accuracy',
      metric_value=logs['val_accuracy'],
      global_step=epoch
    )
custom_callback = CustomCallback()

model.fit(train_dataset, epochs=args.epochs, validation_data=val_dataset, callbacks=[custom_callback])