Class: Trainers::TrainerCallback

Inherits:
Object
  • Object
show all
Defined in:
lib/trainers/callbacks.rb

Overview

Base class for trainer callbacks. Override any hook you need.

Direct Known Subclasses

EarlyStoppingCallback, PrinterCallback

Instance Method Summary collapse

Instance Method Details

#on_epoch_begin(args, state, control, **kwargs) ⇒ Object 



38
 
# File 'lib/trainers/callbacks.rb', line 38

def on_epoch_begin(args, state, control, **kwargs); end

#on_epoch_end(args, state, control, **kwargs) ⇒ Object 



39
 
# File 'lib/trainers/callbacks.rb', line 39

def on_epoch_end(args, state, control, **kwargs); end

#on_evaluate(args, state, control, metrics: nil, **kwargs) ⇒ Object 



43
 
# File 'lib/trainers/callbacks.rb', line 43

def on_evaluate(args, state, control, metrics: nil, **kwargs); end

#on_log(args, state, control, logs: nil, **kwargs) ⇒ Object 



42
 
# File 'lib/trainers/callbacks.rb', line 42

def on_log(args, state, control, logs: nil, **kwargs); end

#on_save(args, state, control, **kwargs) ⇒ Object 



44
 
# File 'lib/trainers/callbacks.rb', line 44

def on_save(args, state, control, **kwargs); end

#on_step_begin(args, state, control, **kwargs) ⇒ Object 



40
 
# File 'lib/trainers/callbacks.rb', line 40

def on_step_begin(args, state, control, **kwargs); end

#on_step_end(args, state, control, **kwargs) ⇒ Object 



41
 
# File 'lib/trainers/callbacks.rb', line 41

def on_step_end(args, state, control, **kwargs); end

#on_train_begin(args, state, control, **kwargs) ⇒ Object 



36
 
# File 'lib/trainers/callbacks.rb', line 36

def on_train_begin(args, state, control, **kwargs); end

#on_train_end(args, state, control, **kwargs) ⇒ Object 



37
 
# File 'lib/trainers/callbacks.rb', line 37

def on_train_end(args, state, control, **kwargs); end