Abdullah-Nazhat commited on
Commit
ae032db
·
verified ·
1 Parent(s): 075a642

Update interactor.py

Browse files
Files changed (1) hide show
  1. 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,score_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)