From 9bae70b22549b68f5cdeece8b6c3b3de00c22714 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 16 Apr 2018 23:51:11 +0000
Subject: [PATCH] Accelerated by another 5% using FP16/32 Batch-norm for Tensor Cores.

---
 cfg/darknet.cfg |   14 +++++++++-----
 1 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index ff0d33e..60b939a 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -11,7 +11,7 @@
 learning_rate=0.1
 policy=poly
 power=4
-max_batches=500000
+max_batches=1600000
 
 [convolutional]
 batch_normalize=1
@@ -84,6 +84,7 @@
 [maxpool]
 size=2
 stride=2
+padding=1
 
 [convolutional]
 batch_normalize=1
@@ -93,12 +94,15 @@
 pad=1
 activation=leaky
 
-[avgpool]
-
-[connected]
-output=1000
+[convolutional]
+filters=1000
+size=1
+stride=1
+pad=1
 activation=leaky
 
+[avgpool]
+
 [softmax]
 groups=1
 

--
Gitblit v1.10.0