diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-04-26 16:33:40 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-04-26 16:33:40 -0700 |
commit | 04273e704592aff0dce9014aae818fe12a4b144a (patch) | |
tree | 6cbdc5b0e4340a792c79f45f75c7562942c1464e /tools/test_net.cpp | |
parent | 8e2a51b8f29daf76b5a24e9af769743e1bf6fdfe (diff) | |
download | caffeonacl-04273e704592aff0dce9014aae818fe12a4b144a.tar.gz caffeonacl-04273e704592aff0dce9014aae818fe12a4b144a.tar.bz2 caffeonacl-04273e704592aff0dce9014aae818fe12a4b144a.zip |
fix test_net to upgrade params from v0 if needed
Diffstat (limited to 'tools/test_net.cpp')
-rw-r--r-- | tools/test_net.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tools/test_net.cpp b/tools/test_net.cpp index 4b7fe8ee..c5819ec7 100644 --- a/tools/test_net.cpp +++ b/tools/test_net.cpp @@ -39,9 +39,7 @@ int main(int argc, char** argv) { } Net<float> 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."; |