bakup dali train val scripts

cnn_e2e/config.py

@@ -27,7 +27,7 @@ def get_parser(parser=None):
     parser.add_argument("--gpu_num_per_node", type=int, default=1)
     parser.add_argument('--num_nodes', type=int, default=1,
                         help='node/machine number for training')
-    parser.add_argument('--node_ips', type=str_list, default=['192.168.1.15', '192.168.1.16'],
+    parser.add_argument('--node_ips', type=str_list, default=['192.168.1.13', '192.168.1.14'],
                         help='nodes ip list for training, devided by ",", length >= num_nodes')
 
     parser.add_argument("--model", type=str, default="vgg16", help="vgg16 or resnet50")

e2e.sh → dali_e2e.sh

@@ -4,7 +4,7 @@ DATA_ROOT=/mnt/13_nfs/xuan/ImageNet/mxnet
   #python3 cnn_benchmark/of_cnn_train_val.py \
 #gdb --args \
 #nvprof -f -o resnet.nvvp \
-  python3 cnn_e2e/of_cnn_train_val.py \
+  python3 cnn_e2e/dali_cnn_train_val.py \
     --data_train=$DATA_ROOT/train.rec \
     --data_train_idx=$DATA_ROOT/train.idx \
     --data_val=$DATA_ROOT/val.rec \