callback

Documentation

Index

• Variables
• type Action
• type Callback
• type Checkpoint
  • func (c *Checkpoint) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)
  • func (c *Checkpoint) GetSaveDir() string
  • func (c *Checkpoint) Init() error
• type CheckpointCompare
• type EarlyStoppingOnMetric
  • func (c *EarlyStoppingOnMetric) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)
  • func (c *EarlyStoppingOnMetric) Init() error
• type EarlyStoppingOnMetricMode
• type Event
• type HasSaveDir
• type Log
• type Logger
  • func (l *Logger) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)
  • func (l *Logger) Init() error
• type Mode
• type RecordStats
  • func (r *RecordStats) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)
  • func (r *RecordStats) Init() error

Variables

var (
	ModeTrain   Mode   = "train"
	ModeVal     Mode   = "val"
	ModeTest    Mode   = "test"
	EventStart  Event  = "start"
	EventDuring Event  = "during"
	EventEnd    Event  = "end"
	EventSave   Event  = "save"
	ActionNop   Action = "nop"
	ActionSave  Action = "save"
	ActionHalt  Action = "halt"
)

var (
	LoggerVerbose      = "verbose"
	LoggerTotalBatches = "totalBatches"
	LoggerPrefetched   = "prefetched"
)

Types

type Action

type Action string

type Callback

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

type Checkpoint

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

func (*Checkpoint) Call

func (c *Checkpoint) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)

func (*Checkpoint) GetSaveDir

func (c *Checkpoint) GetSaveDir() string

func (*Checkpoint) Init

func (c *Checkpoint) Init() error

type CheckpointCompare

type CheckpointCompare string

var (
	CheckpointCompareMin CheckpointCompare = "min"
	CheckpointCompareMax CheckpointCompare = "max"
)

type EarlyStoppingOnMetric

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

func (*EarlyStoppingOnMetric) Call

func (c *EarlyStoppingOnMetric) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)

func (*EarlyStoppingOnMetric) Init

func (c *EarlyStoppingOnMetric) Init() error

type EarlyStoppingOnMetricMode

type EarlyStoppingOnMetricMode string

var (
	Min EarlyStoppingOnMetricMode = "min"
	Max EarlyStoppingOnMetricMode = "max"
)

type Event

type Event string

type HasSaveDir

type HasSaveDir interface {
	GetSaveDir() string
}

type Log

type Log struct {
	Name      string
	Value     float64
	Precision int
}

type Logger

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

func (*Logger) Call

func (l *Logger) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)

func (*Logger) Init

func (l *Logger) Init() error

type Mode

type Mode string

type RecordStats

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

func (*RecordStats) Call

func (r *RecordStats) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error)

func (*RecordStats) Init

func (r *RecordStats) Init() error