from local

This commit is contained in:
dohe0342 2023-04-11 16:03:39 +09:00
parent 7cf6aaa4ef
commit ecb581d7db
2 changed files with 2 additions and 0 deletions

View File

@ -1615,6 +1615,8 @@ def run_adapter(rank, world_size, args, wb=None):
optimizer, scheduler = optimizer_adapter, scheduler_adapter
librispeech = LibriSpeechAsrDataModule(args)
prompt = prompt.to('cuda')
'''
if params.hpo: