Update spanet.py

DoranLyong · May 28, 2024 · 4975998 · 4975998
1 parent 71c5f86
commit 4975998
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/image_classification/spanet.py b/image_classification/spanet.py
@@ -182,7 +182,7 @@ def forward(self, x):
         self.ctx = self.pws[0](feat_bank[0]) + self.pws[1](feat_bank[1]) + self.pws[2](feat_bank[2]) # context aggregation
         x = x * self.ctx # modulation
 
-        # == Aggregation == # 
+        # == Update == # 
         x = self.proj_out(x) 
         return x