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:
- 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,859Updated 2 years ago
- A tool for enriching the output of nvidia-smi.☆575Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Updated last year
- A small package to create visualizations of PyTorch execution graphs☆3,485Updated last year
- Estimates the size of a PyTorch model in memory☆359Updated 5 years ago
- YACS -- Yet Another Configuration System☆1,331Updated 3 years ago
- Make huge neural nets fit in memory☆2,830Updated 5 years ago
- Profiling and inspecting memory in pytorch☆1,077Updated 5 months ago
- A lightweight library for PyTorch training tools and utilities☆1,720Updated this week
- Standalone TFRecord reader/writer with PyTorch data loaders☆900Updated 8 months ago
- PyTorch Boilerplate For Research☆607Updated 4 years ago
- Code snippets created for the PyTorch discussion board☆572Updated 4 years ago
- A learning rate range test implementation in PyTorch☆997Updated 7 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,169Updated last year
- Debug PyTorch code using PySnooper☆800Updated 4 years ago
- PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com☆705Updated 3 years ago
- View model summaries in PyTorch!☆2,901Updated 2 weeks ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Updated 5 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,222Updated 3 weeks ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,754Updated 3 weeks ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,161Updated last year
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Updated 4 years ago
- Deep Learning Experiment Management☆642Updated last month
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,582Updated last year
- Pytorch Lightning code guideline for conferences☆1,286Updated 2 years ago
- Some custom dataset examples for PyTorch☆874Updated 5 years ago
- pip install antialiased-cnns to improve stability and accuracy☆1,680Updated last year
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,614Updated 4 years ago
- Train AI models efficiently on medical images using any framework☆1,877Updated last year