Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
dcuai
dlexamples
Commits
eaff6662
Commit
eaff6662
authored
Aug 01, 2022
by
qianyj
Browse files
update TF code
parent
441c8f40
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
9 deletions
+9
-9
TensorFlow/ComputeVision/Accuracy_Validation/README.md
TensorFlow/ComputeVision/Accuracy_Validation/README.md
+1
-1
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
+8
-8
No files found.
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
# 多卡测试 (单精度)
# 多卡测试 (单精度)
...
...
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
View file @
eaff6662
...
@@ -6,25 +6,25 @@
...
@@ -6,25 +6,25 @@
## 运行
## 运行
export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
export HIP_VISIBLE_DEVICES=0
export HIP_VISIBLE_DEVICES=0
python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1 --use_synthetic_data=false
python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1 --use_synthetic_data=false
# 单卡测试 (混合精度)
# 单卡测试 (混合精度)
## 运行
## 运行
export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
export HIP_VISIBLE_DEVICES=0
export HIP_VISIBLE_DEVICES=0
python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1 --use_synthetic_data=false --dtype=fp16
python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1 --use_synthetic_data=false --dtype=fp16
# 多卡测试 (单精度)
# 多卡测试 (单精度)
## 运行
## 运行
mpirun -np $np --hostfile hostfile --bind-to none scripts-run/single_process.sh
mpirun -np $np --hostfile hostfile --bind-to none scripts-run/single_process.sh
多卡测试时需要修改部分代码,具体可参考[https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en]
多卡测试时需要修改部分代码,具体可参考[https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en]
# 参考
# 参考
[
https://github.com/tensorflow/models/tree/v2.7.0
](
https://github.com/tensorflow/models/tree/v2.7.0/official/vision/image_classification/resnet
)
[
https://github.com/tensorflow/models/tree/v2.7.0
](
https://github.com/tensorflow/models/tree/v2.7.0/official/vision/image_classification/resnet
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment