modlee.exp_loss_logger module

Log several different losses for an experiment.

modlee.exp_loss_logger.extract_loss_functions(code_str: str)

Extracts unique loss function calls from ModleeModel class definition passed as string

Parameters:

code_str – The Python code string to analyze.

Returns:

A list of unique loss function calls found in the code.

modlee.exp_loss_logger.get_exp_loss_for_model(module)

Get a list of loss functions used in a given model. NOTE Assumption that loss function is called in __init__ or training_step

Parameters:

module – The model to analyze.

Returns:

A list of unique loss function calls found in the model.