From 880cf187d87c904f5fe574802ecff99118643f2d Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 09 Mar 2018 16:44:46 +0000
Subject: [PATCH] Fixed multi-GPU training for Tensor Cores

---
 build/darknet/x64/resnet152_yolo.cfg |   30 +++++++-----------------------
 1 files changed, 7 insertions(+), 23 deletions(-)

diff --git a/build/darknet/x64/resnet152_yolo.cfg b/build/darknet/x64/resnet152_yolo.cfg
index 1178d32..0ea6112 100644
--- a/build/darknet/x64/resnet152_yolo.cfg
+++ b/build/darknet/x64/resnet152_yolo.cfg
@@ -5,22 +5,17 @@
 width=416
 channels=3
 momentum=0.9
-decay=0.0001
-#decay=0.0005
+decay=0.0005
 angle=0
 saturation = 1.5
 exposure = 1.5
 hue=.1
 
-learning_rate=0.01
-max_batches = 90000
+learning_rate=0.0001
+max_batches = 45000
 policy=steps
-steps=60000,80000
-scales=.1,.1
-
-#learning_rate=0.0001
-#steps=100,25000,35000
-#scales=10,.1,.1
+steps=100,25000,35000
+scales=10,.1,.1
 
 [convolutional]
 batch_normalize=1
@@ -1436,7 +1431,7 @@
 [shortcut]
 from=-4
 activation=leaky
-stopbackward=1
+#stopbackward=1
 
 
 
@@ -1450,17 +1445,6 @@
 
 
 [convolutional]
-batch_normalize=1
-size=3
-stride=1
-pad=1
-filters=1024
-activation=leaky
-
-
-
-
-[convolutional]
 size=1
 stride=1
 pad=1
@@ -1477,7 +1461,7 @@
 softmax=1
 jitter=.3
 rescore=1
-focal_loss=1
+#focal_loss=1
 
 object_scale=5
 noobject_scale=1

--
Gitblit v1.10.0