Merge pull request #78 from qingqing01/cifar

Refine dataprovider in image classification

image_classification/README.md

@@ -177,7 +177,7 @@ def initializer(settings, mean_path, is_train, **kwargs):
     }
 
 
-@provider(init_hook=initializer, cache=CacheType.CACHE_PASS_IN_MEM)
+@provider(init_hook=initializer, pool_size=50000)
 def process(settings, file_list):
     with open(file_list, 'r') as fdata:
         for fname in fdata:

image_classification/dataprovider.py

@@ -27,7 +27,7 @@ def initializer(settings, mean_path, is_train, **kwargs):
     }
 
 
-@provider(init_hook=initializer, cache=CacheType.CACHE_PASS_IN_MEM)
+@provider(init_hook=initializer, pool_size=50000)
 def process(settings, file_list):
     with open(file_list, 'r') as fdata:
         for fname in fdata:

image_classification/index.html

@@ -218,7 +218,7 @@ def initializer(settings, mean_path, is_train, **kwargs):
     }
 
 
-@provider(init_hook=initializer, cache=CacheType.CACHE_PASS_IN_MEM)
+@provider(init_hook=initializer, pool_size=50000)
 def process(settings, file_list):
     with open(file_list, 'r') as fdata:
         for fname in fdata:

image_classification/train.sh

@@ -14,15 +14,15 @@
 # limitations under the License.
 set -e
 
-#config=models/resnet.py
-config=models/vgg.py
+#cfg=models/resnet.py
+cfg=models/vgg.py
 output=output
 log=train.log
 
 paddle train \
-    --config=$config \
-    --use_gpu=1 \
-    --trainer_count=4 \
+    --config=$cfg \
+    --use_gpu=true \
+    --trainer_count=1 \
     --log_period=100 \
     --num_passes=300 \
     --save_dir=$output \