def save(self, model_folder): """Saves optimal TFT weights. Args: model_folder: Location to serialze model. """ # Allows for direct serialisation of tensorflow variables to avoid spurious # issue with Keras that leads to different performance evaluation results # when model is reloaded (https://github.com/keras-team/keras/issues/4875). utils.save(tf.keras.backend.get_session(), model_folder, cp_name=self.name, scope=self.name)
def save(self, model_folder): utils.save(tf.keras.backend.get_session(), model_folder, cp_name=self.name, scope=self.name)