Skip to content
This repository was archived by the owner on Nov 16, 2023. It is now read-only.

Latest commit

 

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.