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 ef2887c025621fd8fa9bacef0b405e6a9e797361..e61e6ba1b146cae637b024fbc9349b95ab4231ad 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')