diff --git a/torchreid/models/osnet.py b/torchreid/models/osnet.py index 59d2bda..326a56c 100644 --- a/torchreid/models/osnet.py +++ b/torchreid/models/osnet.py @@ -210,6 +210,8 @@ class OSNet(nn.Module): Reference: - Zhou et al. Omni-Scale Feature Learning for Person Re-Identification. ICCV, 2019. + - Zhou et al. Learning Generalisable Omni-Scale Representations + for Person Re-Identification. arXiv preprint, 2019. """ def __init__(self, num_classes, blocks, layers, channels, feature_dim=512, loss='softmax', IN=False, **kwargs):