from local

This commit is contained in:
dohe0342 2022-12-26 14:42:23 +09:00
parent 93228f7f59
commit 743d355fc9
2 changed files with 2 additions and 2 deletions

View File

@ -154,8 +154,8 @@ class ResidualAdapterModule(nn.Module):
self.adapter_layers = nn.ModuleList(
[build_adapter(embedding_dim, proj_dim) for _ in range(layer_num)]
)
def forward(x):#, layer_id=0):#, layer_id=-1):
def forward(self, x, layer_id=-1):
layer_id = 0
print(x.size())
x = x.transpose(0, 1)