lmflow.utils.test_utils

Functions

compare_model(→ None)

Module Contents

lmflow.utils.test_utils.compare_model(model_ref: torch.nn.Module, model_trained: torch.nn.Module, module_trained: List[str] | None = None) → None