mirror of
https://github.com/wassname/ray.git
synced 2026-07-11 09:30:55 +08:00
Maksim Smolin
3a134c7224
* Start renaming pytorch to torch * Rename PyTorchTrainer to TorchTrainer * Rename PyTorch runners to Torch runners * Finish renaming API * Rename to torch in tests * Finish renaming docs + tests * Run format + fix DeprecationWarning * fix * move tests up * rename Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
10 lines
307 B
Python
10 lines
307 B
Python
from ray.util.sgd.torch import TorchTrainer
|
|
from ray.util.sgd.tf import TFTrainer
|
|
|
|
__all__ = ["TorchTrainer", "TFTrainer"]
|
|
|
|
|
|
def PyTorchTrainer(**kwargs):
|
|
raise DeprecationWarning("ray.util.sgd.pytorch.PyTorchTrainer has been "
|
|
"renamed to ray.util.sgd.torch.TorchTrainer")
|