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

Fixed nan issue for training with CUDNN_HALF=1 by using Tensor Cores

Browse Source
This commit is contained in:
AlexeyAB
2018-12-07 22:40:10 +03:00
parent 21a4ec9390
commit 7c2f302321
9 changed files with 318 additions and 249 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/network.h
View File

@ -42,6 +42,7 @@ typedef struct network{
int *steps;
int num_steps;
int burn_in;
int cudnn_half;
int adam;
float B1;