From 9d23aad8696268e8ce3a94fee9490fd1db000dc8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 31 Dec 2017 17:10:32 +0000
Subject: [PATCH] Added CUDA-streams to Darknet-Yolo forward inference

---
 build/darknet/x64/densenet201_yolo.cfg |   23 ++++++++++++++++++++++-
 1 files changed, 22 insertions(+), 1 deletions(-)

diff --git a/build/darknet/x64/densenet201_yolo.cfg b/build/darknet/x64/densenet201_yolo.cfg
index 5d7a194..3435555 100644
--- a/build/darknet/x64/densenet201_yolo.cfg
+++ b/build/darknet/x64/densenet201_yolo.cfg
@@ -1,7 +1,7 @@
 [net]
 # Training
 batch=64
-subdivisions=16
+subdivisions=32
 
 # Testing
 #batch=1
@@ -1937,6 +1937,27 @@
 layers=-1,-3
 
 
+
+[convolutional]
+batch_normalize=1
+size=1
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+
+[convolutional]
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+
+
+
 [convolutional]
 size=1
 stride=1

--
Gitblit v1.10.0