Set phase to TRAIN when performing backward pass

diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp
index 6448ce6..286de48 100644 (file)
--- a/tools/dump_network.cpp
+++ b/tools/dump_network.cpp
@@ -61,6 +61,7 @@ int main(int argc, char** argv) {
   caffe_net->Forward(input_vec);
   if (argc > 5 && strcmp(argv[5], "1") == 0) {
     LOG(ERROR) << "Performing Backward";
+    Caffe::set_phase(Caffe::TRAIN);
     caffe_net->Backward();
     // Dump the network
     NetParameter output_net_param;