imagenet-multiGPU.torch
Neural network trainer
A toolkit for training neural networks on the ImageNet dataset using multiple GPUs.
an imagenet example in torch.
402 stars
37 watching
158 forks
Language: Lua
last commit: over 7 years ago
Linked from 1 awesome list
Related projects:
Repository | Description | Stars |
---|---|---|
soumith/inception.torch | A Torch implementation of a pre-trained neural network architecture used in image recognition tasks | 66 |
karandwivedi42/imagenet-multigpu.torchnet | A Lua-based implementation of a multi-GPU training framework for image classification using the Alexnet model. | 4 |
torch/nn | An open-source neural network package providing a modular framework for building and training neural networks. | 1,343 |
jhkim89/pyramidnet | A Torch implementation of a novel neural network architecture designed to improve the generalization ability of deep image classification models. | 129 |
soumith/net2net.torch | A tool for dynamically widening and deepening neural networks to scale their capacity | 160 |
soumith/cudnn.torch | Torch7 bindings for NVIDIA cuDNN, providing FFI access to its neural network acceleration features | 417 |
soumith/dcgan.torch | A PyTorch implementation of a deep generative model that can be used to generate images from a dataset. | 1,458 |
torch/nngraph | Graphical computation library for building neural network architectures | 299 |
torch/cunn | A CUDA implementation of neural network modules and related GPU operations | 215 |
soumith/torch-android | A tool to enable Android device deployment and execution of machine learning models using the Torch library. | 275 |
cvondrick/torch-starter | A basic package for training neural networks in Torch7, providing a simplified starting point for developers. | 56 |
soumith/torch-signal | A signal processing toolkit with support for various transforms and analysis methods. | 48 |
tyill/sunnet | A lightweight C++ deep learning library for building and training neural networks. | 61 |
loudinthecloud/dpwa | A distributed learning framework that enables peer-to-peer parameter averaging and asynchronous training of deep neural networks | 53 |
nashory/gans-collection.torch | A collection of Torch implementations for training various types of Generative Adversarial Networks (GANs) | 55 |