Skip to content

Instantly share code, notes, and snippets.

@analyticsindiamagazine

analyticsindiamagazine/Logging.py

Created November 8, 2019 04:56
Show Gist options
  • Save analyticsindiamagazine/7d69559765f0729fefe1656009204664 to your computer and use it in GitHub Desktop.
Save analyticsindiamagazine/7d69559765f0729fefe1656009204664 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Logging.py
#A function to log the training process
def run(run_dir, hparams):
with tf.summary.create_file_writer(run_dir).as_default():
hp.hparams(hparams)
rmse = train_test_model(hparams)
tf.summary.scalar(METRIC_RMSE, rmse, step=10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment