fixed a typo

This commit is contained in:
JinZr 2024-09-05 22:35:57 +08:00
parent 6e4a9ea85a
commit 2df992f98a

View File

@ -568,13 +568,13 @@ def train_one_epoch(
speech_hat_i = speech_hat_i.squeeze(0) speech_hat_i = speech_hat_i.squeeze(0)
speech_i = speech_i.squeeze(0) speech_i = speech_i.squeeze(0)
tb_writer.add_audio( tb_writer.add_audio(
"train/valdi_speech_hat", "train/valid_speech_hat",
speech_hat_i, speech_hat_i,
params.batch_idx_train, params.batch_idx_train,
params.sampling_rate, params.sampling_rate,
) )
tb_writer.add_audio( tb_writer.add_audio(
"train/valdi_speech", "train/valid_speech",
speech_i, speech_i,
params.batch_idx_train, params.batch_idx_train,
params.sampling_rate, params.sampling_rate,