anibali / docker-pytorchLinks
A Docker image for PyTorch
☆993Updated last year
Alternatives and similar repositories for docker-pytorch
Users that are interested in docker-pytorch are comparing it to the libraries listed below
Sorting:
- A tool for enriching the output of nvidia-smi.☆567Updated last year
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,192Updated last year
- YACS -- Yet Another Configuration System☆1,316Updated 3 years ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆888Updated 2 months ago
- Profiling and inspecting memory in pytorch☆1,061Updated 11 months ago
- Estimates the size of a PyTorch model in memory☆358Updated 5 years ago
- Deep Learning Experiment Management☆639Updated 2 years ago
- PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com☆704Updated 2 years ago
- Code snippets created for the PyTorch discussion board☆569Updated 4 years ago
- PyTorch Boilerplate For Research☆608Updated 3 years ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,838Updated last year
- A lightweight library for PyTorch training tools and utilities☆1,701Updated last week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,732Updated last week
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,397Updated 6 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,049Updated last year
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Pytorch Lightning code guideline for conferences☆1,273Updated last year
- Debug PyTorch code using PySnooper☆797Updated 4 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆905Updated last year
- Some custom dataset examples for PyTorch☆868Updated 5 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,151Updated 2 weeks ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,661Updated last month
- Tutorial for building a custom CUDA function for Pytorch☆517Updated 6 years ago
- Make huge neural nets fit in memory☆2,802Updated 5 years ago
- A CV toolkit for my papers.☆2,050Updated 6 months ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,824Updated last month
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,533Updated 4 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,572Updated last year
- On the Variance of the Adaptive Learning Rate and Beyond☆2,552Updated 3 years ago