ddpw 3.0.0

A utility package to scaffold PyTorch's DDP

DDPW

---

The Distributed Data Parallel wrapper (DDPW) is created as a utility package to encapsulate the scaffolding for PyTorch's Distributed Data Parallel.

This code is written in Python. The DDPW documentation contains details on how to use this package.

Overview

Installation

conda install -c tvsujal ddpw # with conda
pip install ddpw # with pip from PyPI

Usage

  from ddpw.platform import PlatformConfig
  from ddpw.artefacts import ArtefactsConfig
  from ddpw.job import JobConfig
  from ddpw.wrapper import Wrapper

  from src import MyTrainer

  p = PlatformConfig(...)
  a = ArtefactsConfig(...)
  t = JobConfig(...)

  d = Wrapper(p, a)
  j = MyTrainer(t)

  d.start(j)