diff options
Diffstat (limited to 'models/bvlc_alexnet')
| -rw-r--r-- | models/bvlc_alexnet/train_val.prototxt | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/models/bvlc_alexnet/train_val.prototxt b/models/bvlc_alexnet/train_val.prototxt index 69b8916d..717b6fa4 100644 --- a/models/bvlc_alexnet/train_val.prototxt +++ b/models/bvlc_alexnet/train_val.prototxt @@ -5,7 +5,8 @@ layers { top: "data" top: "label" data_param { - source: "examples/imagenet/ilsvrc12_train_leveldb" + source: "examples/imagenet/ilsvrc12_train_lmdb" + backend: LMDB batch_size: 256 } transform_param { @@ -21,7 +22,8 @@ layers { top: "data" top: "label" data_param { - source: "examples/imagenet/ilsvrc12_val_leveldb" + source: "examples/imagenet/ilsvrc12_val_lmdb" + backend: LMDB batch_size: 50 } transform_param { |