`torch_mist.utils.freeze`

Module Contents

Functions

`freeze`(→ Any)
`n_trainable_parameters`(→ int)
`is_trainable`(→ bool)

torch_mist.utils.freeze.freeze(function: Any) → Any

torch_mist.utils.freeze.n_trainable_parameters(function: Any) → int

torch_mist.utils.freeze.is_trainable(function: Any) → bool