ml-lab Goto Github PK

pytorch-deep-image-matting

Pytorch implementation of deep image matting

pytorch-deform-conv

PyTorch implementation of Deformable Convolution

pytorch-dnc

Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom

pytorch-dnc-1

Differentiable Neural Computers, Sparse Access Memory and Sparse Differentiable Neural Computers, for Pytorch

pytorch-dpn-pretrained

Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation

pytorch-dppo

Pytorch implementation of Distributed Proximal Policy Optimization: https://arxiv.org/abs/1707.02286

pytorch-dynamic-rnn-decoder-tree

This is code I wrote within less than an hour so as to very roughly draft how I would code a Dynamic RNN Decoder Tree with PyTorch.

pytorch-encoding

PyTorch Deep Texture Encoding Network

pytorch-everybodydancenow

Implementation of Everybody Dance Now by pytorch

pytorch-explain-black-box

PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation

pytorch-faster-rcnn

pytorch-fcn

PyTorch Implementation of Fully Convolutional Networks

pytorch-fid

A Port of Fréchet Inception Distance (FID score) to PyTorch

pytorch-flows

PyTorch implementations of algorithms for density estimation

pytorch-gan

PyTorch implementations of Generative Adversarial Networks.

pytorch-gat

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

pytorch-generative-model-collections

Collection of generative models in Pytorch version.

pytorch-grad-cam

PyTorch implementation of Grad-CAM

pytorch-gradual-warmup-lr

Gradually-Warmup Learning Rate Scheduler for PyTorch

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MixNet, MobileNet-V3/V2/V1, MNASNet, Single-Path NAS, FBNet, and more

pytorch-kaldi

pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.

pytorch-lars

pytorch-lightning

The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate

pytorch-looking-fast-and-slow

PyTorch implementation of Looking Fast and Slow: Memory-Guided Mobile Video Object Detection

pytorch-maddpg

A pytorch implementation of MADDPG (multi-agent deep deterministic policy gradient)

pytorch-made

MADE (Masked Autoencoder Density Estimation) implementation in PyTorch

pytorch-maml

PyTorch implementation of MAML: https://arxiv.org/abs/1703.03400

pytorch-matchnet

PyTorch implementation of the MatchNet generative model

pytorch-meta-optimizer

A PyTorch implementation of Learning to learn by gradient descent by gradient descent

pytorch-mobilenet-from-tf

Mobilenet model converted from tensorflow