frogbox

Frogbox is an opinionated PyTorch machine learning framework built for rapid prototyping and research.

Features

• Experiments are defined using JSON files and support jinja2 templates.
• Flexible event system inspired by Ignite.
• Automatic experiment tracking. Currently only Weights & Biases is supported with other platforms planned.
• CLI tool for easy project management. Just type frogbox project new -t supervised to get started.
• Integrates Accelerate to support automatic mixed precision (AMP) and distributed training.

Installation

pip install git+https://SimonLarsen@github.com/SimonLarsen/frogbox.git@v0.5.2b

Getting started

See documentation.