from local

This commit is contained in:
dohe0342 2023-04-11 16:32:32 +09:00
parent 1ebc50a4ec
commit 0e13fd3209
3 changed files with 2 additions and 0 deletions

View File

@ -1622,6 +1622,8 @@ def run_adapter(rank, world_size, args, wb=None):
librispeech = LibriSpeechAsrDataModule(args)
model.prompt = model.prompt.to('cuda')
'''
if params.hpo:
train_cuts = librispeech.train_clean_10_cuts(option=params.gender)