Skip to content

Instantly share code, notes, and snippets.

@MattKovtun

MattKovtun/train_model.py

Last active July 18, 2018 10:41
Show Gist options
  • Save MattKovtun/ce6e5586d45588f3f4d0e80ae67fceb4 to your computer and use it in GitHub Desktop.
Save MattKovtun/ce6e5586d45588f3f4d0e80ae67fceb4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_model.py
# Your training happens here
# Hundreds and hundreds of layers
# Stacked to create the greatest model ever
# don't forget saving to json format
import tensorflowjs as tfjs
tfjs.converters.save_keras_model(model, "model_js")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment