Update s2_train.py (#1159)

This commit is contained in:
彭震东 2024-06-07 22:03:27 +08:00 committed by GitHub
parent 99f09c8bdc
commit e106a5ee88
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -305,7 +305,7 @@ def train_and_evaluate(
y_lengths,
text,
text_lengths,
) in tqdm(enumerate(train_loader)):
) in enumerate(tqdm(train_loader)):
if torch.cuda.is_available():
spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(
rank, non_blocking=True