Skip to content

GitLab

Commit eaff6662 authored by qianyj's avatar qianyj
Browse files

update TF code

parent 441c8f40
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
TensorFlow/ComputeVision/Accuracy_Validation/README.md
View file @ eaff6662
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# 单卡测试 (混合精度) # 单卡测试 (混合精度)
## 运行 ## 运行
export HIP_VISIBLE_DEVICES=0
python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=parameter_server --print_training_accuracy=true --nodistortions --num_gpus=4 --num_epochs=90 --weight_decay=1e-4 --data_dir=$data_dir_path --use_fp16=True --data_name=imagenet --train_dir=$save_checkpoint_path python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=parameter_server --print_training_accuracy=true --nodistortions --num_gpus=4 --num_epochs=90 --weight_decay=1e-4 --data_dir=$data_dir_path --use_fp16=True --data_name=imagenet --train_dir=$save_checkpoint_path
# 多卡测试 (单精度) # 多卡测试 (单精度)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment