def save2file(self, model: keras.models.Model): model_json = model.to_json() with open( os.path.join(self.context.model_dir, self.context.model_name), 'w') as f: f.write(model_json) model.save_weights( os.path.join(self.context.model_dir, self.context.weight_name))
def callback( self, kmodel: keras.models.Model, epoch: int = None, batch: int = None, ) -> None: print('SAVING: %s' % self._url) kmodel.save_weights(self._url, overwrite=True)