from local
This commit is contained in:
parent
93228f7f59
commit
743d355fc9
Binary file not shown.
@ -154,8 +154,8 @@ class ResidualAdapterModule(nn.Module):
|
|||||||
self.adapter_layers = nn.ModuleList(
|
self.adapter_layers = nn.ModuleList(
|
||||||
[build_adapter(embedding_dim, proj_dim) for _ in range(layer_num)]
|
[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
|
layer_id = 0
|
||||||
print(x.size())
|
print(x.size())
|
||||||
x = x.transpose(0, 1)
|
x = x.transpose(0, 1)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user