Utils

store_variables

 store_variables (pkl_fn:str|pathlib.Path, size:list, reorder:bool,
                  resample:int|list)

Save variable values in a pickle file.

source

load_variables

 load_variables (pkl_fn:(<class'str'>,<class'pathlib.Path'>))

Loads stored variable values from a pickle file.

Args: pkl_fn: File path of the pickle file to be loaded.

Returns: The deserialized value of the pickled data.

source

print_colab_gpu_info

 print_colab_gpu_info ()

Check if we have a GPU attached to the runtime.

source

ModelTrackingCallback

 ModelTrackingCallback (model_name:str, loss_function:str,
                        item_tfms:list[typing.Any], size:list[int],
                        resample:list[float], reorder:bool)

A FastAI callback for comprehensive MLflow experiment tracking.

This callback automatically logs hyperparameters, metrics, model artifacts, and configuration to MLflow during training.

source

MLflowUIManager

 MLflowUIManager ()

Initialize self. See help(type(self)) for accurate signature.