From 6aa140b3c6e9917936f73bd538fce4fe20954233 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Sun, 15 Apr 2018 13:57:05 +0000 Subject: [PATCH] Disable burn_in for FP16 Tensor Cores --- src/cost_layer.h | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/src/cost_layer.h b/src/cost_layer.h index aa4af2f..a692831 100644 --- a/src/cost_layer.h +++ b/src/cost_layer.h @@ -1,7 +1,7 @@ #ifndef COST_LAYER_H #define COST_LAYER_H -#include "params.h" #include "layer.h" +#include "network.h" typedef layer cost_layer; -- Gitblit v1.10.0