Pytorch-benchmark

Latest version: v0.3.6

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

Page 2 of 2

Added
- Memory measurement for each batch size.
- Repeated energy measurement.

Changed
- Number formatting to use u instead of µ.

Added
- Option to redirect info prints.

Fixed
- Added missing with torch.no_grad

Added
- Overloads for benchmark parameters and functions to allow benchmark of custom classes.

Fixed
- GPU compatibility.

Removed
- Carbon-tracker energy measurement. Library is still too immature at this point.