scxpand.util.train_logger

scxpand.util.train_logger#

Functions

has_checkpoint(path)

Classes

TrainLogger(save_path[, resume_exp_path, trial])

class scxpand.util.train_logger.TrainLogger(save_path, resume_exp_path=None, trial=None)#
__init__(save_path, resume_exp_path=None, trial=None)#
add_scalars(scalars, group, global_step, epoch=None, i_batch=None, one_line=False)#
flush()#
get_elapsed_time_in_seconds()#
init_writer(n_epochs, n_train_batches, prm)#
load_best_model(model, device_name=None)#
resume_from_checkpoint(resume_exp_path, model, optimizer=None, lr_scheduler=None, device_name='cuda')#
Return type:

int

save_checkpoint(model, optimizer, lr_scheduler, epoch, model_score, dev_set_metrics=None)#
save_final_summary()#
scxpand.util.train_logger.has_checkpoint(path)#
Contents