anibali / docker-pytorch
A Docker image for PyTorch
☆985Updated last year
Alternatives and similar repositories for docker-pytorch:
Users that are interested in docker-pytorch are comparing it to the libraries listed below
- Code snippets created for the PyTorch discussion board☆561Updated 4 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,109Updated last year
- YACS -- Yet Another Configuration System☆1,311Updated 3 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Pytorch Lightning code guideline for conferences☆1,262Updated last year
- A lightweight library for PyTorch training tools and utilities☆1,694Updated last week
- Debug PyTorch code using PySnooper☆798Updated 4 years ago
- Profiling and inspecting memory in pytorch☆1,057Updated 9 months ago
- Deep Learning Experiment Management☆639Updated 2 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,352Updated 4 months ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,599Updated 3 years ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆883Updated 8 months ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,723Updated this week
- PyTorch layer-by-layer model profiler☆607Updated 3 years ago
- My best practice of training large dataset using PyTorch.☆1,098Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,040Updated last year
- Estimates the size of a PyTorch model in memory☆356Updated 4 years ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,827Updated last year
- Codebase for Image Classification Research, written in PyTorch.☆2,153Updated last year
- On the Variance of the Adaptive Learning Rate and Beyond☆2,547Updated 3 years ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,201Updated last year
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,283Updated last year
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,617Updated 3 years ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,176Updated last year
- PyTorch extensions for high performance and large scale training.☆3,313Updated 2 weeks ago
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,423Updated last week
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆761Updated 4 years ago
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,531Updated 4 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,479Updated last year
- High-level batteries-included neural network training library for Pytorch☆401Updated 3 years ago