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

Loss-graph store automatically (iterations == max_batches) at the end of training

Browse Source
This commit is contained in:
AlexeyAB
2018-04-16 13:09:10 +03:00
parent 6aa140b3c6
commit c52fa47428
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/convolutional_layer.c
View File

@ -248,7 +248,7 @@ void cudnn_convolutional_setup(layer *l, int cudnn_preference)
//printf("Tensor Cores - Backward-filter enabled: l->bf_algo = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD_NONFUSED \n");
if (fw == 2 && bd == 2 && bf == 2) printf("TF ");
else if (fw >= 1 && bd >= 1 && bf >= 1) printf("TH ");
else if (fw == 1 && bd == 1 && bf == 1) printf("TH ");
}
}
#endif