Esempi in Python per HorovodParamsWriter.saveMetadata

Esempio n. 1

File: estimator.py Progetto: zhaoxvdong/horovod

 def saveImpl(self, path):
     # Write the parameters
     HorovodParamsWriter.saveMetadata(
         self.instance,
         path,
         self.sc,
         param_serializer_fn=_torch_param_serialize)

Esempio n. 2

 def saveImpl(self, path):
     keras_utils = self.instance._get_keras_utils()
     # Write the parameters
     HorovodParamsWriter.saveMetadata(
         self.instance,
         path,
         self.sc,
         param_serializer_fn=keras_utils.serialize_param_value)