Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
unit_test.sh 350 Bytes
Newer Older
jjsjann123's avatar
[syncBN] (#48)  
jjsjann123 committed
1
2
3 
python single_gpu_unit_test.py
python -m torch.distributed.launch --nproc_per_node=2 two_gpu_unit_test.py
python -m torch.distributed.launch --nproc_per_node=2 two_gpu_unit_test.py --fp64
root's avatar
adding test_groups.py to unit_test.sh  
root committed
4
5 
#beware, you need a system with at least 4 gpus to test group_size<world_size
python -m torch.distributed.launch --nproc_per_node=4 test_groups.py --group_size=2