callback

type Action string

type Callback interface {
	Init() error
	Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)
}

type Checkpoint struct {
	OnEvent    Event
	OnMode     Mode
	Loss       bool
	MetricName string
	Compare    CheckpointCompare
	SaveDir    string
	// contains filtered or unexported fields
}

type CheckpointCompare string

type EarlyStoppingOnMetric struct {
	OnEvent    Event
	OnMode     Mode
	MetricName string
	Mode       EarlyStoppingOnMetricMode
	MaxValue   float64
	MinValue   float64
}

type EarlyStoppingOnMetricMode string

type Event string

type HasSaveDir interface {
	GetSaveDir() string
}

type Log struct {
	Name      string
	Value     float64
	Precision int
}

type Logger struct {
	FileLogger     *cblog.Logger
	Progress       bool
	ProgressLogDir string
	// contains filtered or unexported fields
}

type Mode string

type RecordStats struct {
	RecordDir      string
	RecordFileName string
	OnEvent        Event
	OnMode         Mode
	// contains filtered or unexported fields
}