from local

This commit is contained in:
dohe0342 2022-12-26 14:21:03 +09:00
parent bf17e501f4
commit 7e696f2be5
2 changed files with 2 additions and 0 deletions

View File

@ -96,6 +96,8 @@ class TransformerEncoderAdapter(TransformerEncoder):
x, (z, lr) = layer(
x, self_attn_padding_mask=padding_mask, need_weights=False,
)
print(x.size())
print(i)
x = self.adapters(x, layer_id=i)
if i >= min_layer: