From f0b8e81ad8f170a11d014df7ac8f1e0b726e73ff Mon Sep 17 00:00:00 2001
From: Simon Layton <slayton58@gmail.com>
Date: Thu, 17 Sep 2015 14:13:33 -0400
Subject: [PATCH] Reshape Layers before calling Forward / Backward

---
 src/caffe/net.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index f1579b85a27..aa266cd73d2 100644
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -553,6 +553,7 @@ Dtype Net<Dtype>::ForwardFromTo(int start, int end) {
   }
   for (int i = start; i <= end; ++i) {
     // LOG(ERROR) << "Forwarding " << layer_names_[i];
+    layers_[i]->Reshape(bottom_vecs_[i], top_vecs_[i]);
     Dtype layer_loss = layers_[i]->Forward(bottom_vecs_[i], top_vecs_[i]);
     loss += layer_loss;
     if (debug_info_) { ForwardDebugInfo(i); }
@@ -617,6 +618,7 @@ void Net<Dtype>::BackwardFromTo(int start, int end) {
   CHECK_LT(start, layers_.size());
   for (int i = start; i >= end; --i) {
     if (layer_need_backward_[i]) {
+      layers_[i]->Reshape(bottom_vecs_[i], top_vecs_[i]);
       layers_[i]->Backward(
           top_vecs_[i], bottom_need_backward_[i], bottom_vecs_[i]);
       if (debug_info_) { BackwardDebugInfo(i); }