From 6b6f0f0ca3193ff6707dd0acac8ccd3e3e3b3873 Mon Sep 17 00:00:00 2001 From: vinjn <vinjn.z@gmail.com> Date: Sat, 26 May 2018 13:18:30 +0000 Subject: [PATCH] classifier.c - set batch=1 in "predict", "demo" and "try" mode. --- 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