From 0b9ab19426e41d0b679b4c4bb80f0d193fde0871 Mon Sep 17 00:00:00 2001
From: thatguymike <mike.houston@gmail.com>
Date: Wed, 2 Dec 2015 20:53:57 -0800
Subject: [PATCH] Port Simon's fix foward

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

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index 25239a82082..cf6f71bd8c0 100644
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -576,6 +576,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); }
@@ -640,6 +641,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); }