Skip to content

GitLab

Unverified Commit 03710112 authored by Bin Lu's avatar Bin Lu Committed by GitHub
Browse files

Update test_train_inference_python.sh

parent 921c24f5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
PTDN/test_train_inference_python.sh
View file @ 03710112
...@@ -259,7 +259,7 @@ else ...@@ -259,7 +259,7 @@ else
fi fi
for autocast in ${autocast_list[*]}; do for autocast in ${autocast_list[*]}; do
if [ ${autocast} = "amp" ]; then if [ ${autocast} = "amp" ]; then
set_amp_config="Gloabl.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True" set_amp_config="Global.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True"
else else
set_amp_config=" " set_amp_config=" "
fi fi
......
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