core.utils.trainutils.write_history

core.utils.trainutils.write_history(
    log_dir,
    epoch,
    train_loss,
    test_loss,
    learning_rates,
    aux=None,
    filename='history.txt',
)

Writes losses and learning rate history to csv file.

Parameters

Name	Type	Description	Default
log_dir		directory containing the history file	required
epoch		epoch	required
train_loss		train_loss of epoch	required
test_loss		test_loss of epoch	required
learning_rates		list of learning rates in epoch	required
aux		list of auxiliary information to be logged	None
filename		name of history file	'history.txt'