From 04273e704592aff0dce9014aae818fe12a4b144a Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Sat, 26 Apr 2014 16:33:40 -0700 Subject: [PATCH] fix test_net to upgrade params from v0 if needed --- tools/test_net.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tools/test_net.cpp b/tools/test_net.cpp index 4b7fe8e..c5819ec 100644 --- a/tools/test_net.cpp +++ b/tools/test_net.cpp @@ -39,9 +39,7 @@ int main(int argc, char** argv) { } Net caffe_test_net(argv[1]); - NetParameter trained_net_param; - ReadProtoFromBinaryFile(argv[2], &trained_net_param); - caffe_test_net.CopyTrainedLayersFrom(trained_net_param); + caffe_test_net.CopyTrainedLayersFrom(argv[2]); int total_iter = atoi(argv[3]); LOG(ERROR) << "Running " << total_iter << " iterations."; -- 2.7.4