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

Accelerated DropBlock on GPU

Browse Source
This commit is contained in:
AlexeyAB
2020-02-22 01:04:55 +03:00
parent 0ee9c8668f
commit 6fb817f68b
4 changed files with 144 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/darknet.h
View File

@ -586,6 +586,8 @@ struct layer {
float * loss_gpu;
float * delta_gpu;
float * rand_gpu;
float * drop_blocks_scale;
float * drop_blocks_scale_gpu;
float * squared_gpu;
float * norms_gpu;