pb

type Layer struct {
	Class  string             `protobuf:"bytes,1,opt,name=class,proto3" json:"class,omitempty"`
	Name   string             `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	Params []*Param           `protobuf:"bytes,3,rep,name=params,proto3" json:"params,omitempty"`
	Args   map[string]float32 `` /* 151-byte string literal not displayed */
	// contains filtered or unexported fields
}

type Net struct {
	Layers    []*Layer   `protobuf:"bytes,1,rep,name=layers,proto3" json:"layers,omitempty"`
	Optimizer *Optimizer `protobuf:"bytes,2,opt,name=optimizer,proto3" json:"optimizer,omitempty"`
	// contains filtered or unexported fields
}

type Optimizer struct {
	Class   string            `protobuf:"bytes,1,opt,name=class,proto3" json:"class,omitempty"`
	Options []byte            `protobuf:"bytes,2,opt,name=options,proto3" json:"options,omitempty"`
	Params  []*OptimizerParam `protobuf:"bytes,3,rep,name=params,proto3" json:"params,omitempty"`
	// contains filtered or unexported fields
}

type OptimizerParam struct {
	Params []*Param `protobuf:"bytes,1,rep,name=params,proto3" json:"params,omitempty"`
	// contains filtered or unexported fields
}

type Param struct {
	Type      uint32  `protobuf:"varint,1,opt,name=type,proto3" json:"type,omitempty"`
	ElemCount int64   `protobuf:"varint,2,opt,name=elem_count,json=elemCount,proto3" json:"elem_count,omitempty"`
	Name      string  `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
	Shapes    []int64 `protobuf:"varint,4,rep,packed,name=shapes,proto3" json:"shapes,omitempty"`
	File      string  `protobuf:"bytes,5,opt,name=file,proto3" json:"file,omitempty"`
	// contains filtered or unexported fields
}