mirror of
https://github.com/wassname/ray.git
synced 2026-08-03 13:10:57 +08:00
[tune] Use public methods for trainable (#9184)
This commit is contained in:
@@ -96,6 +96,8 @@ def make_checkpoint_dir(step=None):
|
||||
|
||||
tune.report(hello="world", ray="tune")
|
||||
|
||||
.. warning:: Do not call this function within the Trainable Class API.
|
||||
|
||||
Args:
|
||||
step (int): Current training iteration - used for setting
|
||||
an index to uniquely identify the checkpoint.
|
||||
@@ -138,6 +140,8 @@ def save_checkpoint(checkpoint):
|
||||
|
||||
analysis = tune.run(run_me)
|
||||
|
||||
.. warning:: Do not call this function within the Trainable Class API.
|
||||
|
||||
Args:
|
||||
**kwargs: Any key value pair to be logged by Tune. Any of these
|
||||
metrics can be used for early stopping or optimization.
|
||||
|
||||
Reference in New Issue
Block a user