From 5e448c00f32e80e13eead72318047ce8368aac9a Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 05 Mar 2018 16:59:29 +0000
Subject: [PATCH] Added Focal loss. Use focal_loss=1 in the [region] layer in your cfg-file. alpha = 0.5, gamma = 2

---
 src/dropout_layer_kernels.cu |    1 -
 1 files changed, 0 insertions(+), 1 deletions(-)

diff --git a/src/dropout_layer_kernels.cu b/src/dropout_layer_kernels.cu
index f65e3c3..7e51bd5 100644
--- a/src/dropout_layer_kernels.cu
+++ b/src/dropout_layer_kernels.cu
@@ -6,7 +6,6 @@
 #include "dropout_layer.h"
 #include "cuda.h"
 #include "utils.h"
-#include "params.h"
 }
 
 __global__ void yoloswag420blazeit360noscope(float *input, int size, float *rand, float prob, float scale)

--
Gitblit v1.10.0