diff options
| author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-09-21 15:32:03 -0700 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-09-21 22:07:44 -0700 |
| commit | 73f1de4b9c6b21b894639f75fb4438fa2e61a052 (patch) | |
| tree | 82adba7f092275a86b576c59835d9f911f9b4efd /models | |
| parent | a6693464787b63d805f077961cb39eb96698807f (diff) | |
| download | caffeonacl-73f1de4b9c6b21b894639f75fb4438fa2e61a052.tar.gz caffeonacl-73f1de4b9c6b21b894639f75fb4438fa2e61a052.tar.bz2 caffeonacl-73f1de4b9c6b21b894639f75fb4438fa2e61a052.zip | |
switch examples to lmdb (except for custom data loaders)
Diffstat (limited to 'models')
| -rw-r--r-- | models/bvlc_alexnet/train_val.prototxt | 6 | ||||
| -rw-r--r-- | models/bvlc_reference_caffenet/train_val.prototxt | 6 |
2 files changed, 8 insertions, 4 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 { diff --git a/models/bvlc_reference_caffenet/train_val.prototxt b/models/bvlc_reference_caffenet/train_val.prototxt index d6d64073..073d8aef 100644 --- a/models/bvlc_reference_caffenet/train_val.prototxt +++ b/models/bvlc_reference_caffenet/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 { |