Skip to content

Instantly share code, notes, and snippets.

@yueyericardo

yueyericardo/load_model.py

Created February 20, 2019 17:58
Show Gist options
  • Save yueyericardo/642daae6b4584cb5362b586df015367a to your computer and use it in GitHub Desktop.
Save yueyericardo/642daae6b4584cb5362b586df015367a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_model.py
def loadmodel():
H_model = tf.keras.models.load_model('./models/H_tf.model', custom_objects={'AniCELU':AniCELU})
C_model = tf.keras.models.load_model('./models/C_tf.model', custom_objects={'AniCELU':AniCELU})
N_model = tf.keras.models.load_model('./models/N_tf.model', custom_objects={'AniCELU':AniCELU})
O_model = tf.keras.models.load_model('./models/O_tf.model', custom_objects={'AniCELU':AniCELU})
modelist = [H_model, C_model, N_model, O_model]
return modelist
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment