mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
Merge branch 'k2-fsa:master' into dev/speechllm
This commit is contained in:
commit
d1a535dc76
@ -383,3 +383,7 @@ Please see: [:
|
||||
type=str2bool,
|
||||
default=False,
|
||||
help="""
|
||||
Whether to adapt. If true, we will mix 5% of the new data
|
||||
with 95% of the original data to fine-tune. This is useful
|
||||
Whether to adapt. If true, we will mix 5%% of the new data
|
||||
with 95%% of the original data to fine-tune. This is useful
|
||||
if you want to maintain the performance on the original domain
|
||||
""",
|
||||
)
|
||||
@ -1134,7 +1134,7 @@ def train_one_epoch(
|
||||
f"Epoch {params.cur_epoch}, "
|
||||
f"batch {batch_idx}, loss[{loss_info}], "
|
||||
f"tot_loss[{tot_loss}], batch size: {batch_size}, "
|
||||
f"lr: {cur_lr:.2e}, "
|
||||
f"lr: {cur_lr: .2e}, "
|
||||
+ (f"grad_scale: {scaler._scale.item()}" if params.use_fp16 else "")
|
||||
)
|
||||
|
||||
|
@ -186,7 +186,7 @@ class AttributeDict(dict):
|
||||
tmp = {}
|
||||
for k, v in self.items():
|
||||
# PosixPath is ont JSON serializable
|
||||
if isinstance(v, pathlib.Path) or isinstance(v, torch.device):
|
||||
if isinstance(v, (pathlib.Path, torch.device, torch.dtype)):
|
||||
v = str(v)
|
||||
tmp[k] = v
|
||||
return json.dumps(tmp, indent=indent, sort_keys=True)
|
||||
|
Loading…
x
Reference in New Issue
Block a user