Skip to content

Commit

Permalink
Add horovod estimator save/load/get_models/shutdown (intel-analytics#…
Browse files Browse the repository at this point in the history
…2702)

* add get_model/save/load/shutdown for pytorch horovod estimator

* change args

* remove condition

* change name

* remove get multiple models

* add pytorch estimator ut and disable for now

* meet comments
  • Loading branch information
shanyu-sys committed Aug 12, 2020
1 parent 2e8919a commit 9e9cf5e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/orca/test/dev/run-pytests-ray
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,8 @@ then
fi

echo "Running orca learn ray tests"
python -m pytest -v ../test/zoo/orca/learn/ray
python -m pytest -v ../test/zoo/orca/learn/ray \
--ignore=../test/zoo/orca/learn/ray/pytorch/test_pytorch_estimator.py
exit_status_2=$?
if [ $exit_status_2 -ne 0 ];
then
Expand Down

0 comments on commit 9e9cf5e

Please sign in to comment.