コード例 #1
0
ファイル: estimator.py プロジェクト: zhaoxvdong/horovod 
 def saveImpl(self, path):
     # Write the parameters
     HorovodParamsWriter.saveMetadata(
         self.instance,
         path,
         self.sc,
         param_serializer_fn=_torch_param_serialize)
コード例 #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)