mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +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, def _is_lr_update_step(self,
group: dict, 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 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 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) return (step >= zero_step + cur_update_period)
def _update_lrs(self, def _update_lrs(self,
group: dict, group: dict,
p: Tensor, p: Tensor,