DLWheel

A lightweight deep learning library.

Installation

Editable Installation

pip install -e .

Stable Release

pip install dlwheel

Getting Started

Basic Usage

from pprint import pprint

import dlwheel

cfg = dlwheel.setup()
pprint(cfg)

Backup System

Use the --backup flag to enable automatic backup (stored in ./log by default):

python main.py --backup

Configuration System

YAML Config File (config/default.yaml):

# config/default.yaml
lr: 1e-3
batch_size: 32

Command-Line Argument Override

python main.py --config=config/exp.yaml --name=exp batch_size=2 lr=0.1