matthewdeng
014a9959f1
Revert "[train] add TorchTensorboardProfilerCallback ( #21864 )" ( #22117 )
...
This reverts commit f064306de9
2022-02-04 08:54:16 -08:00
matthewdeng
f064306de9
[train] add TorchTensorboardProfilerCallback ( #21864 )
...
Implement a TorchTensorboardProfilerCallback and corresponding TorchWorkerProfiler to support distributed PyTorch Profiler With TensorBoard integration.
2022-02-03 19:28:12 -08:00
Junwen Yao
eb8adc6105
[train] add a utility function to turn off TF autosharding ( #21887 )
...
This PR adds a utility function to turn off TF autosharding as a temporary solution.
Closes #19324 .
2022-01-28 16:09:06 -08:00
matthewdeng
8119b62640
[train] refactor callback logdir and results preprocessors ( #21468 )
...
* [train] Add TorchTensorboardProfilerCallback and introduce ResultsPreprocessors
* simplify profiler
* read on get_and_clear_profile_traces
* refactor callbacks
* remove var
* Update python/ray/train/callbacks/logging.py
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
* Update python/ray/train/callbacks/results_prepocessors/keys.py
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
* address comments; add tests
* fix test
* address comments
* docs
* address comments'
* fix test
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
2022-01-21 17:23:34 -08:00
xwjiang2010
9af8f11191
Revert "[docs] Clean up doc structure (first part) ( #21667 )" ( #21763 )
...
This reverts commit 38e46c9fb3
2022-01-20 15:30:56 -08:00
Max Pumperla
38e46c9fb3
[docs] Clean up doc structure (first part) ( #21667 )
2022-01-20 16:19:04 +01:00
Balaji Veeramani
7efe1bef11
[Train] Add PrintCallback ( #21261 )
...
Co-authored-by: Amog Kamsetty <amogkamsetty@yahoo.com>
2022-01-03 14:03:04 -08:00
Amog Kamsetty
57db4640ca
[Train] [Tune] Refactor MLflow ( #20802 )
...
Pulls out Tune's MLflow logging logic to a shared MLflow util.
Adds an MLflow logger callback to Ray Train
Closes #20642
2021-12-21 17:17:52 -08:00
Amog Kamsetty
c03b937b95
[Train] Minor migration guide update ( #20683 )
...
* update docs
* tf
2021-11-29 12:42:28 -08:00
Amog Kamsetty
a74cf7ff1c
[Train] Torch Prepare utilities ( #20254 )
...
* update
* formatting
* fix failures
* fix session tests
* address comments
* add to api docs
* package refactor
* wip
* wip
* wip
* finish
* finish
* fix
* comment
* fix
* install horovod for docs
* address comment
* Update python/ray/train/session.py
Co-authored-by: matthewdeng <matthew.j.deng@gmail.com>
* Update python/ray/train/torch.py
Co-authored-by: matthewdeng <matthew.j.deng@gmail.com>
* address comments
* try fix docs
* fix doc build failure
* fix
* fix
* fix
* try fix doc highlighting
* fix docs
Co-authored-by: matthewdeng <matthew.j.deng@gmail.com>
2021-11-15 07:34:17 -08:00
Amog Kamsetty
65a17da2ec
[Train] Refactor Backends ( #20312 )
...
* wip
* finish
* comment
* fix
* install horovod for docs
* address comment
* fix doc build failure
2021-11-13 11:05:53 -08:00
matthewdeng
4674c78050
[Train] Rename Ray SGD v2 to Ray Train ( #19436 )
2021-10-18 22:27:46 -07:00
