예제 #1
0
 def __init__(self, *args, **kws):
     KerasModelCheckpoint.__init__(self, *args, **kws)
     if self.filepath.startswith("gs://"):
         self.on_epoch_end = self._gcp_on_epoch_end
         self._original_filepath = self.filepath
         self._temp_file = tempfile.NamedTemporaryFile()
         self.filepath = self._temp_file.name
예제 #2
0
 def __init__(self, org_model=None, *args, **kwargs):
     ModelCheckpoint.__init__(self, *args, **kwargs)
     self.org_model = org_model
     self.last_file = None