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

Add some visualizations tools #12

Open cezbloch opened 1 month ago

cezbloch commented 1 month ago

Add possibility to visually compare runs on eg. cuda and cpu.