From eb9c88ef734d693e65ec35036811363a35e6b5d3 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Sat, 14 Apr 2018 22:51:21 +0000 Subject: [PATCH] Fixed bug in Tensor Cores V100 (1. Desc in Batch norm, 2. Manually selected algo). Also fixed time measure on Linux for multi-threading. --- src/deconvolutional_layer.h | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/src/deconvolutional_layer.h b/src/deconvolutional_layer.h index 74498c7..2d36e02 100644 --- a/src/deconvolutional_layer.h +++ b/src/deconvolutional_layer.h @@ -2,10 +2,10 @@ #define DECONVOLUTIONAL_LAYER_H #include "cuda.h" -#include "params.h" #include "image.h" #include "activations.h" #include "layer.h" +#include "network.h" typedef layer deconvolutional_layer; -- Gitblit v1.10.0