英文:
How to surpress Completed Stage: Warm Up and Completed Stage: Collection logs from Pytorch Profiler?
问题
当在Jupyter笔记本中运行此代码时，控制台会打印以下内容：
STAGE:2023-05-10 16:41:42 3392672:3392672 ActivityProfilerController.cpp:294] Completed Stage: Warm UpSTAGE:2023-05-10 16:41:42 3392672:3392672 ActivityProfilerController.cpp:300] Completed Stage: Collection
有没有方法来抑制这些警告消息？
英文:
Basically title. When running this code in a Jupyter notebook:
import torchimport torchvision.models as modelsfrom torch.profiler import profile, ProfilerActivitymodel = models.resnet18()inputs = torch.randn(5, 3, 224, 224)with profile(activities=[ProfilerActivity.CPU], record_shapes=True) as prof:model(inputs)
the console prints
STAGE:2023-05-10 16:41:42 3392672:3392672 ActivityProfilerController.cpp:294] Completed Stage: Warm UpSTAGE:2023-05-10 16:41:42 3392672:3392672 ActivityProfilerController.cpp:300] Completed Stage: Collection
Is there a way to suppress these warning messages?
Thanks
答案1
得分: 1
这只是最近在一个拉取请求中修复/添加的。
现在你可以设置环境变量 KINETO_LOG_LEVEL。
例如，在一个bash脚本中：export KINETO_LOG_LEVEL=3
根据源代码，日志级别如下：
enum LoggerOutputType {VERBOSE = 0,INFO = 1,WARNING = 2,ERROR = 3,STAGE = 4,ENUM_COUNT = 5};
至少按照这个问题的说法，日志级别的更改尚未合并。
英文:
This was just recently fixed/added in a pull request
now you can set the env variable KINETO_LOG_LEVEL.
For example in a bash script: export KINETO_LOG_LEVEL=3
The levels according to the source code are:
enum LoggerOutputType {VERBOSE = 0,INFO = 1,WARNING = 2,ERROR = 3,STAGE = 4,ENUM_COUNT = 5};
Thats atleast how it should work, according to this issue the changes for the log level have not been merged yet
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论