diff --git a/docs/source/framework/callbacks.rst b/docs/source/framework/callbacks.rst index acbce497e7..6131acfc22 100644 --- a/docs/source/framework/callbacks.rst +++ b/docs/source/framework/callbacks.rst @@ -30,3 +30,17 @@ We offer several pre-written callbacks which are ready to be used out of the box TorchSnapshotSaver TQDMProgressBar TrainProgressMonitor + +.. fbcode:: + + .. currentmodule:: torchtnt.meta.framework.callbacks + + .. autosummary:: + :nosignatures: + :toctree: generated/ + :template: class_template.rst + + BaseHiveWriter + ModelStoreCheckpointer + StuckJobDetector + ZoomerProfiler diff --git a/docs/source/index.rst b/docs/source/index.rst index 117db32875..1848ea8a7e 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -78,15 +78,6 @@ Documentation utils/data utils/loggers -.. fbcode:: - - .. toctree:: - :maxdepth: 2 - :caption: Framework (Meta) - :glob: - - meta/framework/callbacks - .. fbcode:: .. toctree::