from local
This commit is contained in:
parent
74f51ba091
commit
1817b0a18e
Binary file not shown.
@ -40,7 +40,7 @@ class TransformerEncoderAdapter(TransformerEncoder):
|
|||||||
self.adapters = ResidualAdapterModule()
|
self.adapters = ResidualAdapterModule()
|
||||||
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')/100.)
|
p.data = nn.Parameter(torch.randn(p.size()).to('cuda'))
|
||||||
print(p)
|
print(p)
|
||||||
|
|
||||||
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user