from local
This commit is contained in:
parent
6b13f61fc2
commit
2cf703ffef
Binary file not shown.
@ -43,7 +43,7 @@ class TransformerEncoderAdapter(TransformerEncoder):
|
|||||||
|
|
||||||
for p in self.adapters.parameters():
|
for p in self.adapters.parameters():
|
||||||
# p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
# p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
||||||
p.data = nn.Parameter(torch.randn(p.size()).to('cuda')/50.)
|
p.data = nn.Parameter(torch.randn(p.size()).to('cuda')/100.)
|
||||||
|
|
||||||
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
||||||
x, layer_results = self.extract_features_with_adapter(
|
x, layer_results = self.extract_features_with_adapter(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user