From ec68838342b42541776607b0c14e40fb89f7e3d8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 23 May 2018 15:27:18 +0000
Subject: [PATCH] Fixed memory leaks for Yolo: train, test

---
 build/darknet/x64/resnet152_yolo.cfg |   34 +++++++++-------------------------
 1 files changed, 9 insertions(+), 25 deletions(-)

diff --git a/build/darknet/x64/resnet152_yolo.cfg b/build/darknet/x64/resnet152_yolo.cfg
index 1178d32..d766084 100644
--- a/build/darknet/x64/resnet152_yolo.cfg
+++ b/build/darknet/x64/resnet152_yolo.cfg
@@ -1,26 +1,21 @@
 [net]
 batch=64
-subdivisions=8
+subdivisions=32
 height=416
 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
@@ -1486,4 +1470,4 @@
 
 absolute=1
 thresh = .6
-random=0
+random=1

--
Gitblit v1.10.0