-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Open
Labels
docsDocumentation relatedDocumentation relatedhelp wantedOpen to be worked onOpen to be worked onprofiler
Description
📚 Documentation
Hello,
I was reading about lightning profiling which looks useful and especially how it can be linked with the pytorch profiler.
However, in the doc ( https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.profilers.PyTorchProfiler.html#pytorchprofiler ), there an argument called **profiler_kwargs with doc: Keyword arguments for the PyTorch profiler. This depends on your PyTorch version.
Apart from the fact that this is vague, These kwargs contain most of the important function information you would expect from a profiler:
- use_cpu:bool=True to say that we want to monitor the CPU ,
- use_cuda to indicate whether you want to monitor cuda via the profiler.
Metadata
Metadata
Assignees
Labels
docsDocumentation relatedDocumentation relatedhelp wantedOpen to be worked onOpen to be worked onprofiler