diff --git a/docs/pkg/models.rst b/docs/pkg/models.rst index c8110ea8d..e078444f9 100644 --- a/docs/pkg/models.rst +++ b/docs/pkg/models.rst @@ -38,4 +38,5 @@ ReID-specific Models .. autoclass:: torchreid.models.resnetmid.ResNetMid .. autoclass:: torchreid.models.hacnn.HACNN .. autoclass:: torchreid.models.pcb.PCB -.. autoclass:: torchreid.models.mlfn.MLFN \ No newline at end of file +.. autoclass:: torchreid.models.mlfn.MLFN +.. autoclass:: torchreid.models.osnet.OSNet \ No newline at end of file