From 5ab9a84dbd00283f417854b9e53aca848416a285 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=87=8C?= Date: Wed, 22 Feb 2023 02:58:29 +0000 Subject: [PATCH] update PyTorch/dev/nlp/FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 王凌 --- .../FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/dev/nlp/FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py b/PyTorch/dev/nlp/FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py index ef2887c025..e61e6ba1b1 100644 --- a/PyTorch/dev/nlp/FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py +++ b/PyTorch/dev/nlp/FairSeq_Transformer_ID0496_for_PyTorch/fairseq/options.py @@ -226,7 +226,7 @@ def get_parser(desc, default_task='translation'): help='path to a python module containing custom extensions (tasks and/or architectures)') parser.add_argument('--empty-cache-freq', default=0, type=int, help='how often to clear the PyTorch CUDA cache (0 to disable)') - parser.add_argument('--profiling', type=str, default='NONE', + parser.add_argument('--profiling', type=str, default='CANN', help='choose profiling way--CANN,GE,NONE') parser.add_argument('--start_step', default=0, type=int, help='start_step') -- Gitee