szagoruyko

pytorchviz

A small package to create visualizations of PyTorch execution graphs

attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

wide-residual-networks

3.8% and 18.3% on CIFAR-10 and CIFAR-100

functional-zoo

PyTorch and Tensorflow functional model definitions

diracnets

Training Very Deep Neural Networks Without Skip-Connections

loadcaffe

Load Caffe networks in Torch7

cvpr15deepcompare

Code and models for "Learning to Compare Image Patches via Convolutional Neural Networks"

pyinn

CuPy fused PyTorch neural networks ops

torch-opencv-demos

Torch7+OpenCV+ConvNets

binary-wide-resnet

PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)

imagine-nn

IMAGINE torch neural network routines

nnpack.torch

Torch FFI-bindings for NNPACK

openai-gemm.pytorch

PyTorch bindings for openai-gemm

cutorch-rtc

lua apply function for cutorch

functional-style-transfer

minimal implementation of style transfer

imi-demos

live convolutional neural networks demos

examples

quantile-regression-dqn-pytorch

Quantile Regression DQN a Minimal Working Example

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

cupy

NumPy-like API accelerated with CUDA

deep-symmetry

detectron2

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts

hub

multipathnet

A Torch implementation of the object detection network from "A MultiPath Network for Object Detection" (https://arxiv.org/abs/1604.02135)

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch.github.io

The website for PyTorch

szagoruyko.github.io

tnt

an abstraction to train neural networks

vision

Datasets, Transforms and Models specific to Computer Vision