frogbox

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

Features

• Experiments are defined using YAML or JSON files and support jinja2 templates.
• Flexible event system inspired by Ignite.
• Automatic experiment tracking. Currently, Weights & Biases and MLFlow is supported with other platforms planned.
• CLI tool for easy project management.
• Integrates Accelerate to support automatic mixed precision (AMP) and distributed training.

Installation

pip install frogbox

Getting started

See documentation.