diff --git a/examples/materials/combined/run.sh b/examples/materials/combined/run.sh index 7adf25d..9658f42 100755 --- a/examples/materials/combined/run.sh +++ b/examples/materials/combined/run.sh @@ -1,2 +1,2 @@ export PYTHONPATH=../../../src/:$PYTHONPATH -python3 fit.py +torchrun fit.py diff --git a/examples/synthetic/run.sh b/examples/synthetic/run.sh index ca70a4f..ea93ee8 100755 --- a/examples/synthetic/run.sh +++ b/examples/synthetic/run.sh @@ -1,2 +1,2 @@ export PYTHONPATH=../../src/:$PYTHONPATH -python3 fit.py +torchrun fit.py