elektronn3.training.train_utils module

class elektronn3.training.train_utils.AccumulationArray(right_shape=(), dtype=<class 'numpy.float32'>, n_init=100, data=None, ema_factor=0.95)[source]

Bases: object

add_offset(off)[source]
append(data)[source]
clear()[source]
property data
property ema
max()[source]
mean()[source]
min()[source]
sum()[source]
class elektronn3.training.train_utils.HistoryTracker[source]

Bases: object

plot(save_path=None, autoscale=True, close=True, loss_smoothing_len=200)[source]
register_debug_output_names(names)[source]
update_debug_outputs(vals)[source]
update_history(vals)[source]
update_regression(pred, target)[source]
update_timeline(vals)[source]
class elektronn3.training.train_utils.Timer[source]

Bases: object

property t_passed
elektronn3.training.train_utils.pretty_string_time(t)[source]

Custom printing of elapsed time