diff --git a/docs/api/models.rst b/docs/api/models.rst index ed1b40a37..a925c17e8 100644 --- a/docs/api/models.rst +++ b/docs/api/models.rst @@ -31,3 +31,8 @@ Random-convolutional feature (RCF) extractor ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: RCF + +Residual Network (ResNet) +^^^^^^^^^^^^^^^^^^^^^^^^^ + +.. autofunction:: resnet50 diff --git a/docs/conf.py b/docs/conf.py index 4f0d7706c..41a0e00f1 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -59,6 +59,8 @@ nitpick_ignore = [ ("py:class", ".."), # TODO: can't figure out why this isn't found ("py:class", "LightningDataModule"), + # Undocumented class + ("py:class", "torchvision.models.resnet.ResNet"), ]