Skip to content

GitLab

Commit 598fbc88 authored by root's avatar root
Browse files

adding test_groups.py to unit_test.sh

parent f5725555
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/synced_batchnorm/unit_test.sh
View file @ 598fbc88
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
#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
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