Double aux_grad scale

This commit is contained in:
Daniel Povey 2022-11-28 00:19:03 +08:00
parent 9752778ee6
commit a3b07fd098

View File

@ -343,7 +343,7 @@ def _whitening_schedule(x: float, ratio: float = 2.0) -> ScheduledFloat:
default=x)
def _aux_grad_scale() -> float:
return 0.1
return 0.2
def _aux_grad_prob_out() -> ScheduledFloat:
return ScheduledFloat((0.0, 0.25), (1000.0, 0.05), (8000.0, 0.0125))
def _aux_grad_prob_in() -> ScheduledFloat: