Update interactor.py
Browse files- interactor.py +1 -1
interactor.py
CHANGED
|
@@ -30,7 +30,7 @@ class MemoryUnit(nn.Module):
|
|
| 30 |
return x
|
| 31 |
|
| 32 |
class InteractionUnit(nn.Module):
|
| 33 |
-
def __init__(self,dim
|
| 34 |
super().__init__()
|
| 35 |
|
| 36 |
self.norm_token = nn.LayerNorm(dim)
|
|
|
|
| 30 |
return x
|
| 31 |
|
| 32 |
class InteractionUnit(nn.Module):
|
| 33 |
+
def __init__(self,dim):
|
| 34 |
super().__init__()
|
| 35 |
|
| 36 |
self.norm_token = nn.LayerNorm(dim)
|