pb

type Dense struct {
	Rows int32     `protobuf:"varint,1,opt,name=rows,proto3" json:"rows,omitempty"`
	Cols int32     `protobuf:"varint,2,opt,name=cols,proto3" json:"cols,omitempty"`
	Data []float64 `protobuf:"fixed64,3,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

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 map[string]*Dense `` /* 153-byte string literal not displayed */
	Args   map[string]*Dense `` /* 149-byte string literal not displayed */
	// contains filtered or unexported fields
}

type Loss struct {
	Name   string             `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Params map[string]float64 `` /* 155-byte string literal not displayed */
	// contains filtered or unexported fields
}

type Model struct {
	Name       string     `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	TrainCount uint64     `protobuf:"varint,2,opt,name=train_count,json=trainCount,proto3" json:"train_count,omitempty"`
	ParamCount uint64     `protobuf:"varint,3,opt,name=param_count,json=paramCount,proto3" json:"param_count,omitempty"`
	Layers     []*Layer   `protobuf:"bytes,4,rep,name=layers,proto3" json:"layers,omitempty"`
	Loss       *Loss      `protobuf:"bytes,5,opt,name=loss,proto3" json:"loss,omitempty"`
	Optimizer  *Optimizer `protobuf:"bytes,6,opt,name=optimizer,proto3" json:"optimizer,omitempty"`
	// contains filtered or unexported fields
}

type Optimizer struct {
	Name        string             `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Lr          float64            `protobuf:"fixed64,2,opt,name=lr,proto3" json:"lr,omitempty"`
	WeightDecay float64            `protobuf:"fixed64,3,opt,name=weight_decay,json=weightDecay,proto3" json:"weight_decay,omitempty"`
	Params      map[string]float64 `` /* 155-byte string literal not displayed */
	// contains filtered or unexported fields
}