mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Added assisted_excitation=1 for [convolutional] layer on GPU
This commit is contained in:
@ -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;
|
||||
|
Reference in New Issue
Block a user