anibali / docker-pytorchLinks
A Docker image for PyTorch
☆987Updated 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:
- YACS -- Yet Another Configuration System☆1,315Updated 3 years ago
- A lightweight library for PyTorch training tools and utilities☆1,698Updated this week
- Codebase for Image Classification Research, written in PyTorch.☆2,154Updated last year
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,573Updated 11 months ago
- Estimates the size of a PyTorch model in memory☆357Updated 5 years ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,833Updated last year
- Debug PyTorch code using PySnooper☆798Updated 4 years ago
- Deep Learning Experiment Management☆639Updated 2 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,044Updated last year
- Pytorch Lightning code guideline for conferences☆1,266Updated last year
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,531Updated 4 years ago
- A learning rate range test implementation in PyTorch☆965Updated last month
- A tool for enriching the output of nvidia-smi.☆564Updated last year
- A small package to create visualizations of PyTorch execution graphs☆3,364Updated 5 months ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,730Updated this week
- Code snippets created for the PyTorch discussion board☆563Updated 4 years ago
- A CV toolkit for my papers.☆2,049Updated 5 months ago
- Train AI models efficiently on medical images using any framework☆1,870Updated 11 months ago
- An end-to-end PyTorch framework for image and video classification☆1,603Updated 11 months ago
- Profiling and inspecting memory in pytorch☆1,060Updated 10 months ago
- PyTorch layer-by-layer model profiler☆607Updated 4 years ago
- PyTorch to Keras model convertor☆860Updated 2 years ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,044Updated 3 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,133Updated last week
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,187Updated last year
- Tutorial for building a custom CUDA function for Pytorch☆516Updated 6 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,550Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- Automatic architecture search and hyperparameter optimization for PyTorch☆2,450Updated last year