From fdcc096694791f92d2cff0e4fd74fbabfca12936 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 24 Jul 2015 22:14:23 +0000 Subject: [PATCH] Fixed some problems with visualizing networks --- 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