Pytorchures

Latest version: v0.1.2

Safety actively analyzes 681812 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

0.1.2

Synchronization after Module timing for any accelerator.

Previously only CUDA was synchronized. Now CPU, CUDA and XPU are synchronized, so that profiling timings are correct.

0.1.1b

Additions:

Change the way models are wrapped - now it is done by the TimedLayer constructor.
Add possibility to save profiling data as dictionary and dump to file as json.
store profiling results of multiple runs
store information on which device is each layer location - eg. cpu or cuda

0.0.1

Measure time and print it into logs for every model layer

Links

Releases

Has known vulnerabilities

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.