Browse Source

profiler: optimize the logging for device memory profiling

tags/v1.2.0-rc1
zhangyunshu 4 years ago
parent
commit
95d2656447
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      mindspore/profiler/parser/memory_usage_parser.py

+ 3
- 3
mindspore/profiler/parser/memory_usage_parser.py View File

@@ -61,9 +61,9 @@ class MemoryUsageParser:
file_path = validate_and_normalize_path(file_path)

if not os.path.exists(file_path):
msg = 'The memory file does not exist!'
logger.error(msg)
raise ProfilerFileNotFoundException(msg=msg)
logger.warning('The memory file does not exist! Please ignore the warning '
'if you are running heterogeneous training.')
raise ProfilerFileNotFoundException(msg=file_path)

return file_path



Loading…
Cancel
Save