net

Documentation

Index

• Constants
• func OpenModel(url string) (io.ReadCloser, error)
• func ReadModelWeights(recv func() (*aggregatorpb.ModelWeightChunk, error)) *io.PipeReader
• type ModelWeightsWriter
  • func NewModelWeightsWriter(send func(aggregatorpb.ModelWeightChunk) error) *ModelWeightsWriter
  • func (w *ModelWeightsWriter) Close() error
  • func (w *ModelWeightsWriter) Write(b []byte) (int, error)

Constants

const ModelWeightChunkSize = 64 * units.KB

Functions

func OpenModel

func OpenModel(url string) (io.ReadCloser, error)

OpenModel opens the model for reading.

func ReadModelWeights

func ReadModelWeights(recv func() (*aggregatorpb.ModelWeightChunk, error)) *io.PipeReader

Types

type ModelWeightsWriter

type ModelWeightsWriter struct {
	// contains filtered or unexported fields
}

ModelWeightsWriter writes ModelWeightChunks.

func NewModelWeightsWriter

func NewModelWeightsWriter(send func(aggregatorpb.ModelWeightChunk) error) *ModelWeightsWriter

func (*ModelWeightsWriter) Close

func (w *ModelWeightsWriter) Close() error

func (*ModelWeightsWriter) Write

func (w *ModelWeightsWriter) Write(b []byte) (int, error)