From 846b3b4366c8d1a1ab7215db3ce4f4180ea53cb5 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 30 Jul 2015 23:19:14 +0000 Subject: [PATCH] Added COCO, fixed memory leaks --- cfg/yolo.cfg | 11 +++++------ 1 files changed, 5 insertions(+), 6 deletions(-) diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index 8426bf9..ba44c4d 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -1,13 +1,12 @@ [net] -batch=1 -subdivisions=1 +batch=64 +subdivisions=2 height=448 width=448 channels=3 learning_rate=0.01 momentum=0.9 decay=0.0005 -seen = 0 [crop] crop_width=448 @@ -200,6 +199,6 @@ classes=20 coords=4 rescore=0 -joint=1 -objectness = 0 -background=0 +joint=0 +objectness=1 + -- Gitblit v1.10.0