Skip to main content

ART - Actually Robust Training framework - a framework that teaches good practices when training deep neural networks.

Project description

image

ART - Actually Robust Training framework

Tests Docs


ART is a framework that teaches and keeps an eye on good practices when training deep neural networks. It is inspired by a blog post by Andrej Karpathy “A Recipe for Training Neural Networks”. The framework teaches the user how to properly train DNNs by encouraging the user to use built-in mechanisms that ensure the correctness and robustness of the pipeline using easily usable steps. It allows users not only to learn but also to use it in their future projects to speed up model development.

Table of contents:

Installation

To get started, install ART package using:

pip install art-training

Project creation

To use most of art's features we encourage you to create a new folder for your project using the CLI tool:

python -m art.cli create-project my_project_name

This will create a new folder my_project with a basic structure for your project. To learn more about ART we encourage you to read our documentation, and check our tutorials!

Dashboard

After you run some steps you can see compare their execution in the dashboard. To use the dashboard, firstly install required dependencies:

pip install art-training[dashboard]

and run this command in the directory of your project (directory with folder called art_checkpoints).

python -m art.cli run-dashboard

Optionally you can use --experiment-folder switch to pass path to the folder. For more info, use --help switch.

Tutorials

  1. A showcase of ART's features. To check it out type:
python -m art.cli get-started

and launch tutorial.ipynb

After running all cells run dashboard with

python -m art.cli run-dashboard
  1. A tutorial showing how to use ART for transfer learning in an NLP task.
python -m art.cli bert-transfer-learning-tutorial

Required knowledge

In order to use ART, you should have a basic knowledge of:

  • Python - you can find many tutorials online, e.g. here
  • Basic knowledge of machine learning & neural networks - you can find many tutorials online, e.g. here
  • PyTorch - you can find many tutorials online, e.g. here
  • PyTorch Lightning - you can find many tutorials online, e.g. here

Contributing

We welcome contributions to ART! Please check out our contributing guide

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

art_training-0.2.tar.gz (95.4 kB view hashes)

Uploaded Source

Built Distribution

art_training-0.2-py3-none-any.whl (29.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page