Esempi in Python per models.Model.save_weights

Esempio n. 1

File: keras_wrapper.py Progetto: yoraba/fx_rnn

 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))

Esempio n. 2

File: kerashelper.py Progetto: johnathanlouie/albumsummarizer

 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)