/dl_ext

Deep Learning extensions for python and pytorch.

Primary LanguageJupyter Notebook

dl-ext

This repository contains several useful utils for os, file, pytorch manipulation.

Install

This repository has been release in pypi.
Firstly, install pytorch. We strongly recommend to use cuda 9.0, 9.2 or 10.0. Then install dl-ext using 'pip install dl-ext'

Documentation

Documentation has not been edited. easy training with distributed example