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 multi-GPU training for Tensor Cores

Browse Source
This commit is contained in:
AlexeyAB
2018-03-09 19:44:46 +03:00
parent a6c51e3b75
commit 880cf187d8
5 changed files with 48 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/network.h
View File

@ -64,6 +64,10 @@ typedef struct network{
#ifdef GPU
float **input_gpu;
float **truth_gpu;
float **input16_gpu;
float **output16_gpu;
size_t *max_input16_size;
size_t *max_output16_size;
int wait_stream;
#endif
} network;