fastbnns.utils package

Submodules

fastbnns.utils.torch_utils module

Miscellaneous utility functions useful throughout repository.

fastbnns.utils.torch_utils.set_requires_grad_(module: Module, requires_grad: bool, tag: str) → None

Set requires_grad property of all parameters whose name contains tag.

Module contents