add decoder

0756ce54 · ouyangyu · 6a2e74fa · 0756ce54 · 0756ce54
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

Classification/cnns/train_fp16.sh Classification/cnns/train_fp16.sh +1 -0

Classification/cnns/train_fp32.sh Classification/cnns/train_fp32.sh +1 -0

未找到文件。
--- a/Classification/cnns/train_fp16.sh
+++ b/Classification/cnns/train_fp16.sh
@@ -46,6 +46,7 @@ python3 of_cnn_train_val.py \
     --fuse_bn_add_relu=True \
     --nccl_fusion_threshold_mb=16 \
     --nccl_fusion_max_ops=24 \
+     --bytes_decoder=True \
     --num_epoch=$NUM_EPOCH \
     --model="resnet50" 2>&1 | tee ${LOGFILE}


--- a/Classification/cnns/train_fp32.sh
+++ b/Classification/cnns/train_fp32.sh
@@ -44,6 +44,7 @@ python3 of_cnn_train_val.py \
     --fuse_bn_add_relu=True \
     --nccl_fusion_threshold_mb=16 \
     --nccl_fusion_max_ops=24 \
+     --bytes_decoder=True \
     --num_epoch=$NUM_EPOCH \
     --model="resnet50" 2>&1 | tee ${LOGFILE}