Update interactor.py

interactor.py

@@ -5,7 +5,7 @@ from torch import nn
        
  
 
-class MemoryUnit(nn.Module):
+class MappingUnit(nn.Module):
     def __init__(self,dim):
         super().__init__()
         
@@ -55,7 +55,7 @@ class InteractorBlock(nn.Module):
         super().__init__()
        
          
-        self.memory = MemoryUnit(d_model)
+        self.mapping = MappingUnit(d_model)
         self.interaction = InteractionUnit(d_model)
         
     def forward(self, x):
@@ -68,7 +68,7 @@ class InteractorBlock(nn.Module):
         
         residual = x
         
-        x = self.memory(x)
+        x = self.mapping(x)
         
                                           
         out = x + residual