mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Added assisted_excitation=1 for [convolutional] layer on GPU

Browse Source
This commit is contained in:
AlexeyAB
2019-09-03 01:35:05 +03:00
parent a63782ca89
commit be5d0d6693
8 changed files with 232 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/darknet.h
View File

@ -537,6 +537,9 @@ struct layer {
float * rand_gpu;
float * squared_gpu;
float * norms_gpu;
float *gt_gpu;
float *a_avg_gpu;
#ifdef CUDNN
cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
cudnnTensorDescriptor_t srcTensorDesc16, dstTensorDesc16;