nlp-recipes/utils_nlp/common
Daisy Deng 055645f590 fix move and parallelized model for DistributedDataParallel models 2020-03-20 19:18:29 +00:00
..
README.md
__init__.py
pytorch_utils.py fix move and parallelized model for DistributedDataParallel models 2020-03-20 19:18:29 +00:00
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.