mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
undo docs change
This commit is contained in:
1 parent
a0c38e8430
commit
c9151d4a90
21 files changed
+7547
-7568
No files matched your search
@@ -23,24 +23,7 @@ class RayGenerator(nn.Module):
|
||||
...
|
||||
```
|
||||
|
||||
Alternatively, you can also time parts of the code:
|
||||
```python
|
||||
...
|
||||
|
||||
def forward(self, ray_indices: TensorType["num_rays", 3]) -> RayBundle:
|
||||
# implementation here
|
||||
with profiler.time_function("code1"):
|
||||
# some code here
|
||||
...
|
||||
|
||||
with profiler.time_function("code2"):
|
||||
# some code here
|
||||
...
|
||||
...
|
||||
```
|
||||
|
||||
|
||||
At termination of training or end of the training run, the profiler will print out the average execution time for all of the functions or code blocks that have the profiler tag.
|
||||
At termination of training or end of the training run, the profiler will print out the average execution time for all of the functions that have the profiler tag.
|
||||
|
||||
:::{admonition} Tip
|
||||
:class: info
|
||||
@@ -49,14 +32,6 @@ Use this profiler if there are *specific/individual functions* you want to measu
|
||||
:::
|
||||
|
||||
|
||||
#### Profiling with PyTorch profiler
|
||||
|
||||
If you want to profile the training or evaluation code and track the memory and CUDA kernel launches, consider using [PyTorch profiler](https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html).
|
||||
It will run the profiler for some selected step numbers, once with `CUDA_LAUNCH_BLOCKING=1`, once with `CUDA_LAUNCH_BLOCKING=0`.
|
||||
The PyTorch profiler can be enabled with `--logging.profiler=pytorch` flag.
|
||||
The outputs of the profiler are trace files stored in `{PATH_TO_MODEL_OUTPUT}/profiler_traces`, and can be loaded in Google Chrome by typing `chrome://tracing`.
|
||||
|
||||
|
||||
#### Profiling with PySpy
|
||||
|
||||
If you want to profile the entire codebase, consider using [PySpy](https://github.com/benfred/py-spy).
|
||||
|
||||
Reference in new issue
Block a user