Skip to content

GitLab

Commit 1c664582 authored by lcskrishna's avatar lcskrishna
Browse files

Merge branch 'master' of https://github.com/ROCmSoftwarePlatform/apex into ifu_07272020

parents 6f7a8b39 8dd19e3b
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 10 deletions
tests/distributed/run_rocm_distributed.sh
View file @ 1c664582
......@@ -18,16 +18,16 @@ rm rank1master.pth
# Test with opt_level="O5"
echo "running opt_level O5"
python3.6 -m torch.distributed.launch --nproc_per_node=2 amp_master_params/amp_master_params.py --opt_level "O5"
python3.6 amp_master_params/compare.py
# delete the model files
echo "O5 test completed. Deleting model files"
rm rank0model.pth
rm rank1model.pth
rm rank0master.pth
rm rank1master.pth
#echo "running opt_level O5"
#python3.6 -m torch.distributed.launch --nproc_per_node=2 amp_master_params/amp_master_params.py --opt_level "O5"
#python3.6 amp_master_params/compare.py
#
## delete the model files
#echo "O5 test completed. Deleting model files"
#rm rank0model.pth
#rm rank1model.pth
#rm rank0master.pth
#rm rank1master.pth
## Run the Sync BN Tests.
echo "Running syncbn tests"
......
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