Skip to content
Machine Learning research with PyTorch
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
torchzoo
.gitignore
Pipfile
Pipfile.lock
README.md
setup.py

README.md

TorchZoo

Implementations of various NNs in pytorch

# simple models
from torchzoo import FeedForward, COOL

# recurrent models
from torchzoo import RWA

Benchmarks / Reproduction studies are held in TorchCircus.

How to use Zoo/Circus

  1. Finding a paper
    • all book keeping is done in the github Issues.
    • Since most papers end up on arxiv, search for an arxiv ID like 1805.10369
    • If the paper is not on Arxiv, search for the full title of the paper.
  2. The issue labels provide a mechanism to find specific things.
  3. If a paper passes it's reproduction in the Circus, the underlying CORE parts are put into the master branch.
  4. If a paper has failed in it's reproduction, the core parts are sent to the fertilizer branch.
  5. If something is still not covered, please open an issue.

I'll try to look into things whenever I have time.

You can’t perform that action at this time.