diff options
| author | Yangqing Jia <jiayq@google.com> | 2014-07-26 08:37:05 -0700 |
|---|---|---|
| committer | Yangqing Jia <jiayq@google.com> | 2014-07-26 08:37:05 -0700 |
| commit | 9862d5a4dcc70edba94291a64c303b659e7bd362 (patch) | |
| tree | c006bc0c3b11842c1176b1e864944042139a1bad /examples/imagenet | |
| parent | 70b01f2164dd5e53605fc58c25530da3a5cccbd3 (diff) | |
| download | caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.tar.gz caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.tar.bz2 caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.zip | |
LOG(ERROR)->LOG(FATAL), and misc script changes.
Diffstat (limited to 'examples/imagenet')
| -rwxr-xr-x | examples/imagenet/resume_training.sh | 5 | ||||
| -rwxr-xr-x | examples/imagenet/train_alexnet.sh | 3 | ||||
| -rwxr-xr-x | examples/imagenet/train_imagenet.sh | 3 |
3 files changed, 7 insertions, 4 deletions
diff --git a/examples/imagenet/resume_training.sh b/examples/imagenet/resume_training.sh index 2b3b4031..b43a9b1b 100755 --- a/examples/imagenet/resume_training.sh +++ b/examples/imagenet/resume_training.sh @@ -2,7 +2,8 @@ TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin \ - imagenet_solver.prototxt caffe_imagenet_train_10000.solverstate +GLOG_logtostderr=1 $TOOLS/caffe.bin train\ + --solver_proto_file=imagenet_solver.prototxt \ + --resume_point_file=caffe_imagenet_train_10000.solverstate echo "Done." diff --git a/examples/imagenet/train_alexnet.sh b/examples/imagenet/train_alexnet.sh index c91e9108..d43a235d 100755 --- a/examples/imagenet/train_alexnet.sh +++ b/examples/imagenet/train_alexnet.sh @@ -2,6 +2,7 @@ TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin alexnet_solver.prototxt +GLOG_logtostderr=1 $TOOLS/caffe.bin train \ + --solver_proto_file=alexnet_solver.prototxt echo "Done." diff --git a/examples/imagenet/train_imagenet.sh b/examples/imagenet/train_imagenet.sh index 0d2563b0..90dcdfad 100755 --- a/examples/imagenet/train_imagenet.sh +++ b/examples/imagenet/train_imagenet.sh @@ -2,6 +2,7 @@ TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin imagenet_solver.prototxt +GLOG_logtostderr=1 $TOOLS/caffe.bin train \ + --solver_proto_file=imagenet_solver.prototxt echo "Done." |