Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
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