From 4c8d77d14a0347ec998fc6f3d4a7689ae46bc0bf Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 15 Jul 2022 14:18:07 +0800 Subject: [PATCH] Fix return type --- egs/librispeech/ASR/pruned_transducer_stateless7/optim.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py index 7cd69feb4..de8b8681e 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py @@ -513,7 +513,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of def _is_lr_update_step(self, group: dict, - state: dict) -> False: + state: dict) -> bool: """ Returns True if on this step we need to update the learning-rate matrices for this tensor and False if not. The periodicity with which we update them increases from @@ -534,7 +534,6 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of return (step >= zero_step + cur_update_period) - def _update_lrs(self, group: dict, p: Tensor,