Esempio n. 1
0
 def saveImpl(self, path):
     # Write the parameters
     HorovodParamsWriter.saveMetadata(
         self.instance,
         path,
         self.sc,
         param_serializer_fn=_torch_param_serialize)
Esempio n. 2
0
 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)