LGDec 7, 2014
Theano-based Large-Scale Visual Recognition with Multiple GPUs
arXiv:1412.2302v446 citationsHas Code
Originality Synthesis-oriented
AI Analysis
This provides an open-source Python-based AlexNet implementation, which is incremental as it adapts existing methods to a new framework.
The authors tackled the problem of implementing AlexNet for large-scale visual recognition using Theano and multiple GPUs, achieving performance on 2 GPUs comparable to Caffe on 1 GPU.
In this report, we describe a Theano-based AlexNet (Krizhevsky et al., 2012) implementation and its naive data parallelism on multiple GPUs. Our performance on 2 GPUs is comparable with the state-of-art Caffe library (Jia et al., 2014) run on 1 GPU. To the best of our knowledge, this is the first open-source Python-based AlexNet implementation to-date.