docs (catalyst-team#1152)

zkid18 · Apr 4, 2021 · fcbcfd1 · fcbcfd1
1 parent 5a15977
commit fcbcfd1
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/docs/api/callbacks.rst b/docs/api/callbacks.rst
@@ -99,6 +99,13 @@ Misc – TqdmCallback
     :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
     :show-inheritance:
 
+OnnxCallback
+~~~~~~~~~~~~
+.. autoclass:: catalyst.callbacks.onnx.OnnxCallback
+    :members:
+    :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
+    :show-inheritance:
+
 OptimizerCallback
 ~~~~~~~~~~~~~~~~~~~~~~
 .. autoclass:: catalyst.callbacks.optimizer.OptimizerCallback
@@ -127,7 +134,12 @@ PruningCallback
     :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
     :show-inheritance:
 
-.. Quantization
+QuantizationCallback
+~~~~~~~~~~~~~~~~~~~~
+.. autoclass:: catalyst.callbacks.quantization.QuantizationCallback
+    :members:
+    :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
+    :show-inheritance:
 
 Scheduler – SchedulerCallback
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -143,7 +155,12 @@ Scheduler – LRFinder
     :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
     :show-inheritance:
 
-.. Tracing
+Tracing
+~~~~~~~
+.. autoclass:: catalyst.callbacks.tracing.TracingCallback
+    :members:
+    :exclude-members: __init__, on_experiment_start, on_stage_start, on_epoch_start, on_loader_start, on_batch_start, on_batch_end, on_loader_end, on_epoch_end, on_stage_end, on_experiment_end
+    :show-inheritance:
 
 
 Metric