From cf36ce977c0b8bd54c906e63ce31931ac060178f Mon Sep 17 00:00:00 2001 From: anyvoice <130631963+anyvoiceai@users.noreply.github.com> Date: 星期日, 02 七月 2023 09:12:18 +0800 Subject: [PATCH] Creates tensor in target device to avoid high CPU occupation. (#695) --- egs/aishell2/transformerLM/conf/train_lm_transformer.yaml | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/egs/aishell2/transformerLM/conf/train_lm_transformer.yaml b/egs/aishell2/transformerLM/conf/train_lm_transformer.yaml index c256295..4155a79 100644 --- a/egs/aishell2/transformerLM/conf/train_lm_transformer.yaml +++ b/egs/aishell2/transformerLM/conf/train_lm_transformer.yaml @@ -13,7 +13,7 @@ batch_type: numel batch_bins: 6000000 accum_grad: 1 -max_epoch: 15 # 15epoch is enougth +max_epoch: 15 # 15epoch is enough optim: adam optim_conf: -- Gitblit v1.9.1