add script

Author: zhengxiawu

README.md

@@ -1,2 +1,9 @@
-# pytorch_image_classification
-image classification by pytroch
+## Requirements
+- Pytorch 1.0
+- Python 3.6+
+- DALI
+```bash
+# Train A network with ImageNet64 memory data by dali
+./script/data_to_memory.sh imagenet64
+python train_dali.py --dataset ImageNet64 --data_path /userhome/temp_data/cifar10 --model_method proxyless_NAS --model_name proxyless_gpu
+```

config.py

@@ -43,7 +43,7 @@ class AugmentConfig(BaseConfig):
     def build_parser(self):
         parser = get_parser("Augment config")
         parser.add_argument('--name', default='')
-        parser.add_argument('--dataset', default='CIFAR10', help='CIFAR10 / MNIST / FashionMNIST')
+        parser.add_argument('--dataset', default='CIFAR10', help='CIFAR10 / ImageNet64 / FashionMNIST')
         parser.add_argument('--data_path', required=False, default='/userhome/data/cifar10',
                             help='data path')
         parser.add_argument('--model_method', default='proxyless_NAS',)

script/data_to_memory.sh

@@ -0,0 +1,14 @@
+#!/usr/bin/env bash
+mount -t tmpfs -o size=30G tmpfs /userhome/temp_data
+if [ "$1" == "imagenet64" ]; then
+cp -r /userhome/data/ImageNet64 /userhome/temp_data/
+elif [ "$1" == 'imagenet32' ];
+then
+cp -r /userhome/data/ImageNet32 /userhome/temp_data/
+elif [ "$1" == 'imagenet16' ];
+then
+cp -r /userhome/data/ImageNet16 /userhome/temp_data/
+elif [ "$1" == 'cifar10' ];
+then
+cp -r /userhome/data/cifar10 /userhome/temp_data/
+fi