Versions in this module Expand all Collapse all v0 v0.26.28 Mar 22, 2022 Changes in this version + type EarlyStoppingOnMetric struct + MaxValue float64 + MetricName string + MinValue float64 + Mode EarlyStoppingOnMetricMode + OnEvent Event + OnMode Mode + func (c *EarlyStoppingOnMetric) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error) + func (c *EarlyStoppingOnMetric) Init() error + type EarlyStoppingOnMetricMode string + var Max EarlyStoppingOnMetricMode = "max" + var Min EarlyStoppingOnMetricMode = "min" type Logger + Progress bool + ProgressLogDir string v0.2.6 Dec 6, 2021 Changes in this version + var ActionHalt Action = "halt" + var ActionNop Action = "nop" + var ActionSave Action = "save" + var EventDuring Event = "during" + var EventEnd Event = "end" + var EventSave Event = "save" + var EventStart Event = "start" + var LoggerPrefetched = "prefetched" + var LoggerTotalBatches = "totalBatches" + var LoggerVerbose = "verbose" + var ModeTest Mode = "test" + var ModeTrain Mode = "train" + var ModeVal Mode = "val" + type Action string + type Callback interface + Call func(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error) + Init func() error + type Checkpoint struct + Compare CheckpointCompare + Loss bool + MetricName string + OnEvent Event + OnMode Mode + SaveDir string + 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 string + var CheckpointCompareMax CheckpointCompare = "max" + var CheckpointCompareMin CheckpointCompare = "min" + type Event string + type HasSaveDir interface + GetSaveDir func() string + type Log struct + Name string + Precision int + Value float64 + type Logger struct + FileLogger *cblog.Logger + func (l *Logger) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error) + func (l *Logger) Init() error + type Mode string + type RecordStats struct + OnEvent Event + OnMode Mode + RecordDir string + RecordFileName string + func (r *RecordStats) Call(event Event, mode Mode, epoch int, batch int, logs []Log) ([]Action, error) + func (r *RecordStats) Init() error
Go to main page