From 3c405fb6fdd461711181ce3dd5df71e6150e3583 Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Fri, 6 May 2022 15:37:30 +0800 Subject: [PATCH] minor fix --- egs/librispeech/ASR/pruned_transducer_stateless4/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless4/train.py b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py index e3121935a..2f2fbd838 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless4/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py @@ -868,7 +868,7 @@ def run(rank, world_size, args): model.device = device if rank == 0: - model_avg = model_avg.to(torch.device("cpu")) + model_avg.to(torch.device("cpu")) model_avg.device = device optimizer = Eve(model.parameters(), lr=params.initial_lr)