Pytorch-benchmark

Latest version: v0.3.6

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

Page 1 of 2

Fixed
- Updated on-gpu model benchmaking with best-practices on `cuda.Event` and `cuda.synchronize`.
- FLOPs measurement error on CUDA.

Added
- Repo DOI

Fixed
- Add missing memory to results.

Fixed
- Memory measurement for bs=1.

Fixed
- Warm up batch size.

Removed
- `try_custom_warmup`.

Added
- `warm_up_fn` overload option.
- Support for FLOPs count in torch.nn.Module with input other than Tensor.

Has known vulnerabilities