from local

This commit is contained in:
dohe0342 2022-12-30 11:26:51 +09:00
parent 67a3b79200
commit 5b51a8d74e
3 changed files with 4 additions and 0 deletions

View File

@ -747,6 +747,10 @@ def main():
model.to(device) model.to(device)
model.eval() model.eval()
for n, p in model.named_parameters():
if 'adapter' in n:
print(p)
if "fast_beam_search" in params.decoding_method: if "fast_beam_search" in params.decoding_method:
if params.decoding_method == "fast_beam_search_nbest_LG": if params.decoding_method == "fast_beam_search_nbest_LG":
lexicon = Lexicon(params.lang_dir) lexicon = Lexicon(params.lang_dir)