anibali / docker-pytorch

Related projects ⓘ

Alternatives and complementary repositories for docker-pytorch

• peci1 / nvidia-htop
  A tool for enriching the output of nvidia-smi.
  ☆540Updated 7 months ago
• rbgirshick / yacs
  YACS -- Yet Another Configuration System
  ☆1,277Updated 2 years ago
• vahidk / tfrecord
  Standalone TFRecord reader/writer with PyTorch data loaders
  ☆863Updated 2 months ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,018Updated 3 months ago
• waleedka / hiddenlayer
  Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
  ☆1,796Updated 9 months ago
• pytorch / tnt
  A lightweight library for PyTorch training tools and utilities
  ☆1,663Updated this week
• anderskm / gputil
  A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python
  ☆1,138Updated 6 months ago
• facebookresearch / FixRes
  This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423
  ☆1,032Updated 3 years ago
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,017Updated 8 months ago
• BloodAxe / pytorch-toolbelt
  PyTorch extensions for fast R&D prototyping and Kaggle farming
  ☆1,519Updated 3 weeks ago
• cybertronai / gradient-checkpointing
  Make huge neural nets fit in memory
  ☆2,726Updated 4 years ago
• Lightning-Universe / lightning-transformers
  Flexible components pairing 🤗 Transformers with Pytorch Lightning
  ☆611Updated last year
• Lightning-Universe / lightning-bolts
  Toolbox of models, callbacks, and datasets for AI/ML researchers.
  ☆1,691Updated last week
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆543Updated 3 years ago
• zasdfgbnm / TorchSnooper
  Debug PyTorch code using PySnooper
  ☆802Updated 3 years ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆683Updated 4 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆514Updated 5 years ago
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆639Updated last year
• moemen95 / Pytorch-Project-Template
  A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
  ☆872Updated last year
• facebookresearch / pycls
  Codebase for Image Classification Research, written in PyTorch.
  ☆2,137Updated 7 months ago
• rwightman / gen-efficientnet-pytorch
  Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
  ☆1,566Updated 4 months ago
• szagoruyko / pytorchviz
  A small package to create visualizations of PyTorch execution graphs
  ☆3,219Updated 7 months ago
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,586Updated this week
• lucidrains / lambda-networks
  Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute
  ☆1,532Updated 3 years ago
• kevinzakka / pytorch-goodies
  PyTorch Boilerplate For Research
  ☆603Updated 2 years ago
• zhanghang1989 / PyTorch-Encoding
  A CV toolkit for my papers.
  ☆2,040Updated 5 months ago
• jacobkimmel / pytorch_modelsize
  Estimates the size of a PyTorch model in memory
  ☆352Updated 4 years ago
• nitrain / nitrain
  Train AI models efficiently on medical images using any framework
  ☆1,865Updated 4 months ago
• Lyken17 / Efficient-PyTorch
  My best practice of training large dataset using PyTorch.
  ☆1,087Updated 6 months ago
• kakaobrain / fast-autoaugment
  Official Implementation of 'Fast AutoAugment' in PyTorch.
  ☆1,591Updated 3 years ago