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
488a71ba
Commit
488a71ba
authored
Dec 22, 2022
by
qianyj
Browse files
update tensorflow test file
parent
c056df78
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
TensorFlow/Accuracy_Validation/scripts-run/single_process.sh
TensorFlow/Accuracy_Validation/scripts-run/single_process.sh
+2
-2
TensorFlow/ComputeVision/Classification/scripts-run/single_process.sh
...omputeVision/Classification/scripts-run/single_process.sh
+2
-2
TensorFlow2x/ComputeVision/Classification/scripts-run/single_process.sh
...omputeVision/Classification/scripts-run/single_process.sh
+1
-1
No files found.
TensorFlow/Accuracy_Validation/scripts-run/single_process.sh
View file @
488a71ba
...
...
@@ -4,9 +4,9 @@ lrank=$OMPI_COMM_WORLD_LOCAL_RANK
comm_rank
=
$OMPI_COMM_WORLD_RANK
comm_size
=
$OMPI_COMM_WORLD_SIZE
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
"
case
${
l
_
rank
}
in
case
${
lrank
}
in
[
0]
)
export
HIP_VISIBLE_DEVICES
=
0,1,2,3
export
UCX_NET_DEVICES
=
mlx5_0:1
...
...
TensorFlow/ComputeVision/Classification/scripts-run/single_process.sh
View file @
488a71ba
...
...
@@ -4,9 +4,9 @@ lrank=$OMPI_COMM_WORLD_LOCAL_RANK
comm_rank
=
$OMPI_COMM_WORLD_RANK
comm_size
=
$OMPI_COMM_WORLD_SIZE
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
"
case
${
l
_
rank
}
in
case
${
lrank
}
in
[
0]
)
export
HIP_VISIBLE_DEVICES
=
0,1,2,3
export
UCX_NET_DEVICES
=
mlx5_0:1
...
...
TensorFlow2x/ComputeVision/Classification/scripts-run/single_process.sh
View file @
488a71ba
...
...
@@ -4,7 +4,7 @@ lrank=$OMPI_COMM_WORLD_LOCAL_RANK
comm_rank
=
$OMPI_COMM_WORLD_RANK
comm_size
=
$OMPI_COMM_WORLD_SIZE
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
APP
=
" python3 ./benchmarks-master/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_format=NCHW --batch_size=128 --model=resnet50 --optimizer=momentum --variable_update=horovod --print_training_accuracy=true --nodistortions --num_gpus=1 --num_epochs=90 --weight_decay=1e-4 --data_dir=
$data_dir_path
--use_fp16=False --data_name=imagenet --train_dir=
$save_checkpoint_path
"
case
${
lrank
}
in
[
0]
)
...
...
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