ls_mlkit.util.offload.context module
¶
class
ls_mlkit.util.offload.context.
OffloadContext
(
model
:
Module
,
named_grads
:
dict
,
num_block
=
2
,
no_split_module_classes
=
None
,
enable_gradient_offload
=
True
,
enable_model_offload
=
True
)
[source]
¶
Bases:
object
ls-mlkit
Navigation
Installation:
Installation Guide
API Reference:
ls_mlkit
ls_mlkit package
Subpackages
ls_mlkit.dataset package
ls_mlkit.diffuser package
ls_mlkit.model package
ls_mlkit.optimizer package
ls_mlkit.pipeline package
ls_mlkit.scheduler package
ls_mlkit.util package
Module contents
Related Topics
Documentation overview
ls_mlkit
ls_mlkit package
ls_mlkit.util package
ls_mlkit.util.offload package
Previous:
ls_mlkit.util.offload package
Next:
ls_mlkit.util.offload.forward_backward_offload module