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.