From f7a17f82eb43de864a4f980f235055da9685eef8 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 29 Jan 2014 00:28:42 +0000
Subject: [PATCH] Convolutional layers working w/ matrices

---
 src/parser.c |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/src/parser.c b/src/parser.c
index dc1db2b..eeb6f93 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -107,6 +107,8 @@
         ++count;
         n = n->next;
     }   
+    net.outputs = get_network_output_size(net);
+    net.output = get_network_output(net);
     return net;
 }
 

--
Gitblit v1.10.0