https://github.com/simonlarsen/frogbox

An opinionated machine learning framework.
https://github.com/simonlarsen/frogbox

Last synced: 24 days ago
JSON representation

An opinionated machine learning framework.

• Host: GitHub
• URL: https://github.com/simonlarsen/frogbox
• Owner: SimonLarsen
• License: mit
• Created: 2023-09-01T15:23:29.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-11-05T13:19:25.000Z (about 2 months ago)
• Last Synced: 2024-11-05T14:28:05.655Z (about 2 months ago)
• Language: Python
• Homepage: http://simonlarsen.github.io/frogbox
• Size: 396 KB
• Stars: 1
• Watchers: 3
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        


    





    frogbox



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

## Features

* Built around [Torch Ignite](https://pytorch-ignite.ai) and [Accelerate](https://huggingface.co/docs/accelerate/index) to support automatic mixed precision (AMP) and distributed training.

* Experiments are defined using JSON files and support [jinja2](https://jinja.palletsprojects.com) templates.

* Automatic experiment tracking. Currently only [Weights & Biases](https://wandb.ai/) is supported with other platforms planned.

* CLI tool for easy project management. Just type `frogbox project new -t supervised` to get started.

## Installation

```sh

pip install git+https://[email protected]/SimonLarsen/[email protected]

```

## Getting started

See [documentation](https://simonlarsen.github.io/frogbox).