mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix return type

Browse Source
This commit is contained in:
Daniel Povey 2022-07-15 14:18:07 +08:00
parent 68c5935691
commit 4c8d77d14a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
View File

@ -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,