from local

This commit is contained in:
dohe0342 2023-01-03 16:55:23 +09:00
parent 0a3eb2838d
commit 841d62f98c
2 changed files with 1 additions and 1 deletions

View File

@ -147,7 +147,7 @@ class ResidualAdapterModule(nn.Module):
self.type = 'linear'
def build_adapter(embedding_dim, proj_dim, type_='linear'):
def build_adapter(embedding_dim, proj_dim, type_=self.type):
if type_ == 'conv':
return ConvolutionModule(768, 31)
else: