055645f590 | ||
---|---|---|
.. | ||
README.md | ||
__init__.py | ||
pytorch_utils.py | ||
timer.py |
README.md
Common
This submodule contains high-level common utilities used across multiple algorithms and
frameworks as well as helper functions for managing aspects of different frameworks like pytorch.
For example, pytorch_utils.py contains utilities to interact with PyTorch
like getting a device architecture (cpu or gpu), moving a model to a specific device, and handling
parallelism when multiple gpus are present.