Doc string update for action recognition model

analogdevicesinc · Oct 8, 2024 · 0111857 · 0111857
1 parent 68c625b
commit 0111857
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/models/ai85net-actiontcn.py b/models/ai85net-actiontcn.py
@@ -18,6 +18,9 @@
 class AI85ActionTCN(nn.Module):
     """
     Conv2D backbone + TCN layers for Action Recognition
+    Model was designed to be used with the Kinetics dataset.
+    Number of frames was set to 15, as the model optimally performs with this number
+    within the constraints of the AI85 hardware.
     """
     def __init__(
             self,