From 4785245e5cffaf8b94bc3b51def2e59d939a7244 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 13 Jul 2022 06:51:23 +0800 Subject: [PATCH] Reduce debug freq --- egs/librispeech/ASR/pruned_transducer_stateless7/optim.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py index 5ca634518..009755859 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py @@ -592,7 +592,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of # (where the stats permit). scale = (S.clamp(min=eps) / cur_param_var.clamp(min=eps)).sqrt() - if random.random() < 0.01: + if random.random() < 0.001: logging.info(f"shape={p.shape}, dim={dim}, scale={scale[0].flatten()[::10]}, cur_param_var={cur_param_var[0].flatten()[::10]}, S={S[0].flatten()[::10]}") # scale shape: (batch_size, 1, size, 1, 1) @@ -619,7 +619,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of # cur_scales for the other dims. cur_scales = [None] * ndim - debug = True #(random.random() < 0.001) + debug = (random.random() < 0.001) for i in range(4): # for 4 iterations (this is quite arbitrary) for dim in range(1, ndim): size = p.shape[dim]