From dda993f3dd3c753dfd580d485b39c1001830fee4 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 22 Feb 2018 19:54:40 +0000 Subject: [PATCH] Use half_float16 instead of float32 if defined both CUDNN and CUDNN_HALF. Use Tensor Cores. --- src/dropout_layer.h | 1 - 1 files changed, 0 insertions(+), 1 deletions(-) diff --git a/src/dropout_layer.h b/src/dropout_layer.h index 0c2ce4d..691cfc5 100644 --- a/src/dropout_layer.h +++ b/src/dropout_layer.h @@ -1,7 +1,6 @@ #ifndef DROPOUT_LAYER_H #define DROPOUT_LAYER_H -#include "params.h" #include "layer.h" #include "network.h" -- Gitblit v1.10.0