From 0915e989ca3bf26c3dd2febd4ebb3edec125602a Mon Sep 17 00:00:00 2001
From: Chao Fan <92680886+ChaoFan996@users.noreply.github.com>
Date: Tue, 27 Dec 2022 13:02:32 +0800
Subject: [PATCH] Update modules.py

---
 opengait/modeling/modules.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/opengait/modeling/modules.py b/opengait/modeling/modules.py
index 958a0f2..a911b91 100644
--- a/opengait/modeling/modules.py
+++ b/opengait/modeling/modules.py
@@ -151,6 +151,11 @@ class SeparateBNNecks(nn.Module):
 
 
 class FocalConv2d(nn.Module):
+    """
+        GaitPart: Temporal Part-based Model for Gait Recognition
+        CVPR2020: https://openaccess.thecvf.com/content_CVPR_2020/papers/Fan_GaitPart_Temporal_Part-Based_Model_for_Gait_Recognition_CVPR_2020_paper.pdf
+        Github: https://github.com/ChaoFan96/GaitPart
+    """
     def __init__(self, in_channels, out_channels, kernel_size, halving, **kwargs):
         super(FocalConv2d, self).__init__()
         self.halving = halving