Update interactor.py
Browse files- interactor.py +1 -1
interactor.py
CHANGED
|
@@ -56,7 +56,7 @@ class InteractorBlock(nn.Module):
|
|
| 56 |
|
| 57 |
|
| 58 |
self.memory = MemoryUnit(d_model)
|
| 59 |
-
self.interaction = InteractionUnit(d_model
|
| 60 |
|
| 61 |
def forward(self, x):
|
| 62 |
|
|
|
|
| 56 |
|
| 57 |
|
| 58 |
self.memory = MemoryUnit(d_model)
|
| 59 |
+
self.interaction = InteractionUnit(d_model)
|
| 60 |
|
| 61 |
def forward(self, x):
|
| 62 |
|