Skip to content

Files

Latest commit

Aug 19, 2019
c813dc0 · Aug 19, 2019

History

History
7 lines (6 loc) · 448 Bytes

README.md

File metadata and controls

7 lines (6 loc) · 448 Bytes

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.