ceshine / pytorch-helper-bot

A high-level PyTorch helper package

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Helper Bot

[WIP] a high-level PyTorch helper package

This project is intended for my personal use. Backward compatibility will not be guaranteed. Important releases will be tagged.

Motivation

fast.ai is great, and I recommend it for all deep learning beginners. But since it's beginner-friendly, a lot of more sophisticated stuffs are abstracted heavily and hidden from users. Reading the source code is often required before you can tweak the underlying algorithms. The advent of doc function greatly speeds up the process by quickly directing the user to the source code and documentation.

However, fast.ai has become stronger and bigger. Not everyone has time to keep up with its codebase. Hence the creation of this project. I built a relatively thin layer of abstraction upon PyTorch from scratch, with a lot of ideas and code borrowed from various sources (mainly fast.ai). Only features that are relevant to my use cases are added.

Another similar project is pytorch/ignite.

Tips

  • Set the environment variable SEED, and we will set the random seed of Python, Numpy and PyTorch for you.
  • Set the environment variable DETERMINISTIC to make CuDNN operations deterministic.
  • Set pbar=False when initializing a bot to disable progress bar during evaluation.

Examples

There are almost no unit tests yet. The following example(s) are somewhat functional tests.

About

A high-level PyTorch helper package

License:MIT License


Languages

Language:Python 100.0%