demo

gayeranderson · Sep 28, 2022 · 8bac5fc · 8bac5fc
1 parent 091847d
commit 8bac5fc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -8,3 +8,4 @@
     sa = MobileViTv2Attention(d_model=512)
     output=sa(input)
     print(output.shape)
+