提交 9b1d9969 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add in the doc cuDNN BN and new CPU parallelized operations

上级 0c57414e
...@@ -173,3 +173,14 @@ Softmax Ops ...@@ -173,3 +173,14 @@ Softmax Ops
.. automodule:: theano.sandbox.cuda.dnn .. automodule:: theano.sandbox.cuda.dnn
:noindex: :noindex:
:members: GpuDnnSoftmax, GpuDnnSoftmaxGrad :members: GpuDnnSoftmax, GpuDnnSoftmaxGrad
.. _libdoc_cuda_dnn_bn:
Batch Normalization
===================
.. automodule:: theano.sandbox.cuda.dnn
:noindex:
:members: dnn_batch_normalization_train, dnn_batch_normalization_test
...@@ -10,4 +10,7 @@ ...@@ -10,4 +10,7 @@
.. moduleauthor:: LISA .. moduleauthor:: LISA
.. seealso:: :ref:`cuDNN batch normalization <libdoc_cuda_dnn_bn>`
must be added manually.
.. autofunction:: theano.tensor.nnet.bn.batch_normalization .. autofunction:: theano.tensor.nnet.bn.batch_normalization
...@@ -4,6 +4,13 @@ ...@@ -4,6 +4,13 @@
Multi cores support in Theano Multi cores support in Theano
============================= =============================
Convolution and Pooling
=======================
Since Theano 0.9dev2, the convolution and pooling are parallelized on
CPU.
BLAS operation BLAS operation
============== ==============
......
...@@ -54,7 +54,7 @@ PLATFORMS = ["Windows", "Linux", "Solaris", "Mac OS-X", "Unix"] ...@@ -54,7 +54,7 @@ PLATFORMS = ["Windows", "Linux", "Solaris", "Mac OS-X", "Unix"]
MAJOR = 0 MAJOR = 0
MINOR = 9 MINOR = 9
MICRO = 0 MICRO = 0
SUFFIX = "dev1" # Should be blank except for rc's, betas, etc. SUFFIX = "dev2" # Should be blank except for rc's, betas, etc.
ISRELEASED = False ISRELEASED = False
VERSION = '%d.%d.%d%s' % (MAJOR, MINOR, MICRO, SUFFIX) VERSION = '%d.%d.%d%s' % (MAJOR, MINOR, MICRO, SUFFIX)
......
...@@ -2672,6 +2672,8 @@ def dnn_batch_normalization_train(inputs, gamma, beta, mode='per-activation', ...@@ -2672,6 +2672,8 @@ def dnn_batch_normalization_train(inputs, gamma, beta, mode='per-activation',
Notes Notes
----- -----
Request cuDNN 5 and Theano 0.9dev2 or more recent.
For 4d tensors, returned values are equivalent to: For 4d tensors, returned values are equivalent to:
>>> axes = 0 if mode == 'per-activation' else (0, 2, 3) >>> axes = 0 if mode == 'per-activation' else (0, 2, 3)
...@@ -2736,6 +2738,8 @@ def dnn_batch_normalization_test(inputs, gamma, beta, mean, var, ...@@ -2736,6 +2738,8 @@ def dnn_batch_normalization_test(inputs, gamma, beta, mean, var,
Notes Notes
----- -----
Request cuDNN 5 and Theano 0.9dev2 or more recent.
For 4d tensors, the returned value is equivalent to: For 4d tensors, the returned value is equivalent to:
>>> axes = (0,) if mode == 'per-activation' else (0, 2, 3) >>> axes = (0,) if mode == 'per-activation' else (0, 2, 3)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论