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☆900Updated 8 months ago
- Estimates the size of a PyTorch model in memory☆359Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,719Updated 2 weeks ago
- A tool for enriching the output of nvidia-smi.☆575Updated last year
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,209Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,754Updated last week
- Profiling and inspecting memory in pytorch☆1,077Updated 4 months ago
- YACS -- Yet Another Configuration System☆1,330Updated 3 years ago
- Train AI models efficiently on medical images using any framework☆1,876Updated last year
- A learning rate range test implementation in PyTorch☆997Updated 7 months ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,860Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Updated last year
- Debug PyTorch code using PySnooper☆801Updated 4 years ago
- Pytorch Lightning code guideline for conferences☆1,287Updated 2 years ago
- Deep Learning Experiment Management☆641Updated last month
- Code snippets created for the PyTorch discussion board☆569Updated 4 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,484Updated last year
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,832Updated last week
- PyTorch Boilerplate For Research☆608Updated 3 years ago
- PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com☆704Updated 2 years ago
- Make huge neural nets fit in memory☆2,828Updated 5 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Updated 4 years ago
- PyTorch to Keras model convertor☆863Updated 3 years ago
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,532Updated 5 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Updated 5 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,727Updated last week
- torchbearer: A model fitting library for PyTorch☆641Updated 2 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆912Updated 2 years ago
- Some custom dataset examples for PyTorch☆874Updated 5 years ago
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,582Updated last year