rwightman/pytorch-docker

Stars
2
Language
Python
Created about 7 years ago
Updated about 7 years ago

rwightman/pytorch-docker

rwightman

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Dockerfile definitions with matching Docker Hub images for PyTorch

gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

posenet-python

A Python port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

posenet-pytorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

pytorch-dpn-pretrained

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

udacity-driving-reader

Quick docker based reader for udacity rosbag self-driving dataset. Dumps to png/jpg + csv or Tensorflow examples.

efficientnet-jax

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

pytorch-nips2017-attack-example

A PyTorch baseline attack example for the NIPS 2017 adversarial competition

tensorflow-litterbox

Tensorflow experimentation sandbox. VGG, ResNet, Inception V3, Inception V4, Inception-Resnet-V2 models.

pytorch-pommerman-rl

PyTorch RL for Pommerman

pytorch-planet-amazon

PyTorch models and training code for 'Planet: Understanding the Amazon from Space' Kaggle

pytorch-pretrained-gluonresnet

Well trained MXNet Gluon Model Zoo ResNet/ResNeXt/SE-ResNeXt ported to PyTorch

imagenet-12k

ImageNet-12k subset of ImageNet-21k (fall11)

pytorch-nips2017-adversarial

NIPS 2017 Adversarial Competition in PyTorch

pytorch-opensim-rl

PyTorch based Reinforcement Learning for OpenSim Prosthetics and Learning to Run environments

pytorch-commands

Some PyTorch code for the Kaggle Speech Recognition Challenge

pytorch-nips2017-defense-example

A PyTorch baseline defense example for the NIPS 2017 adversarial competition

pytorch-countception-sealion

Pytorch implementation of Count-ception and custom CNN counting models for Kaggle Sea Lion Count challenge

tensorflow-annex

A module for converting datasets + annotations to sharded Tensorflow TFRecords files

tensorflow-speech_commands

Speech commands training/models from TF repo adapted for speech commands Kaggle

obstacle-tower-pytorch-rainbow

PyTorch & Rainbow for Obstacle Tower Challenge

tensorflow-models-slim

Filtered research/slim folder from tensorflow models repository. With experiments and additions.

boost

My boost library mirror

obstacle-tower-pytorch-a2c-ppo

PPO/A2C in PyTorch for the Obstacle Tower Challenge

ariac-workspace

Docker based setup for experimenting with ARIAC ROS setup

androids-dream

pytorch-cdiscount