imagenet-multiGPU.torch

Neural network trainer

A toolkit for training neural networks on the ImageNet dataset using multiple GPUs.

an imagenet example in torch.

GitHub

402 stars
37 watching
158 forks
Language: Lua
last commit: over 7 years ago
Linked from 1 awesome list


Backlinks from these awesome lists:

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