anibali / docker-pytorchLinks
A Docker image for PyTorch
☆993Updated 2 years ago
Alternatives and similar repositories for docker-pytorch
Users that are interested in docker-pytorch are comparing it to the libraries listed below
Sorting:
- Standalone TFRecord reader/writer with PyTorch data loaders☆899Updated 7 months ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,210Updated last year
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,856Updated last year
- A lightweight library for PyTorch training tools and utilities☆1,718Updated last week
- Profiling and inspecting memory in pytorch☆1,076Updated 3 months ago
- Estimates the size of a PyTorch model in memory☆359Updated 5 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,481Updated last year
- YACS -- Yet Another Configuration System☆1,329Updated 3 years ago
- A tool for enriching the output of nvidia-smi.☆572Updated last year
- Code snippets created for the PyTorch discussion board☆570Updated 4 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,065Updated last year
- Debug PyTorch code using PySnooper☆801Updated 4 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,208Updated 4 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,167Updated last year
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- Deep Learning Experiment Management☆641Updated 3 weeks ago
- PyTorch Boilerplate For Research☆608Updated 3 years ago
- Pytorch Lightning code guideline for conferences☆1,284Updated 2 years ago
- My best practice of training large dataset using PyTorch.☆1,105Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,208Updated 2 years ago
- Tutorial for building a custom CUDA function for Pytorch☆521Updated 6 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,756Updated last month
- Train AI models efficiently on medical images using any framework☆1,876Updated last year
- PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com☆706Updated 2 years ago
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,529Updated 5 years ago
- PyTorch layer-by-layer model profiler☆607Updated 4 years ago
- Make huge neural nets fit in memory☆2,827Updated 5 years ago
- PyTorch to Keras model convertor☆863Updated 3 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,613Updated 4 years ago