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