ls_mlkit.util.offload.model_offload module

class ls_mlkit.util.offload.model_offload.ModelOffloadHookContext(model, no_split_module_classes=None, num_block: int = 2, enable=True, device='cuda', strategy='block', with_backward_hook=False)

Bases: object