AlexeyAB
2018-03-19 537d135feba179636e9bbfe296e078d51f59914c
refs
author AlexeyAB <alexeyab84@gmail.com>
Monday, March 19, 2018 23:16 +0000
committer AlexeyAB <alexeyab84@gmail.com>
Monday, March 19, 2018 23:16 +0000
commit537d135feba179636e9bbfe296e078d51f59914c
tree 2409b5f5fb5814f035ebc8c598fe9c61acfdbb76 tree | zip | gz | tar | bzip2 | xz
parent 2f52cfeb07e11fce8275496164293cf30b732907 view | diff
Improve training performance - batch-norm using cuDNN.
12 files modified
235 ■■■■ changed files
src/batchnorm_layer.c 103 ●●●●● diff | view | raw | blame | history
src/blas.h 1 ●●●● diff | view | raw | blame | history
src/blas_kernels.cu 43 ●●●●● diff | view | raw | blame | history
src/connected_layer.c 19 ●●●●● diff | view | raw | blame | history
src/convolutional_kernels.cu 10 ●●●●● diff | view | raw | blame | history
src/convolutional_layer.c 4 ●●●● diff | view | raw | blame | history
src/cuda.h 32 ●●●●● diff | view | raw | blame | history
src/detector.c 3 ●●●● diff | view | raw | blame | history
src/image.c 1 ●●●● diff | view | raw | blame | history
src/layer.h 13 ●●●●● diff | view | raw | blame | history
src/network_kernels.cu 2 ●●● diff | view | raw | blame | history
src/region_layer.c 4 ●●●● diff | view | raw | blame | history