[Tune] Trainable documentation fix (#9448)

This commit is contained in:
Amog Kamsetty
2020-07-13 13:15:01 -07:00
committed by GitHub
parent e6225bdfa1
commit 4454d05bcf
+1 -1
View File
@@ -175,7 +175,7 @@ class Trainable:
Calling ``save()`` should save the training state of a trainable to disk,
and ``restore(path)`` should restore a trainable to the given state.
Generally you only need to implement ``build``, ``step``,
Generally you only need to implement ``setup``, ``step``,
``save_checkpoint``, and ``load_checkpoint`` when subclassing Trainable.
Other implementation methods that may be helpful to override are