search

cezbloch / pytorchures

Python profiler for pytorch models. Wraps each layer and measures it's execution time.
MIT License
2 stars 0 forks source link

Implement detailed profiling per layer #3

Closed cezbloch closed 3 months ago

cezbloch commented 3 months ago

A wrapper was created for torch.nn.Module layers.

There is a code traversing the model object and wraps each module for profiling.