debug

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

type CSVDebugger struct {
}

type DebugLayer interface {
	weight.Layer

	GetDebugInfo() []*LayerInfo
}

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

type LayerInfo struct {
	ID string

	WeightsImg []image.Image
	BiasImg    []image.Image
	OutImg     []image.Image
	ErrImg     []image.Image

	WeightsStats *tensor.Stats
	BiasStats    *tensor.Stats
	OutStats     *tensor.Stats
	ErrStats     *tensor.Stats

	Other map[string]interface{}
}

type NetDebugger interface {
	Debug(status <-chan string, layerInfo <-chan []*LayerInfo, trainInfo <-chan *TrainInfo, testInfo <-chan *TestInfo)
}

type TestInfo struct {
	Epoch    int
	Loss     float64
	Accuracy float64
}

type TrainInfo struct {
	Epoch             int
	Epochs            int
	Batch             int
	Batches           int
	Loss              float64
	Accuracy          float64
	ExamplesPerSecond float64
}