Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-10-31 09:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
d126ee3053
petals/benchmarks
History
Alexander Borzunov d126ee3053
Add benchmark scripts (#319) 
This PR:

- Adds benchmark scripts for inference, forward pass, and full training step (e.g. used for experiments in our paper).
- Fixes bug with dtypes in `petals.DistributedBloomForSequenceClassification`.
- (minor refactor) Moves `DTYPE_MAP` to `petals.constants` as a useful constant.
2023-06-30 01:12:59 +04:00
..
benchmark_forward.py Add benchmark scripts (#319) 2023-06-30 01:12:59 +04:00
benchmark_inference.py Add benchmark scripts (#319) 2023-06-30 01:12:59 +04:00
benchmark_training.py Add benchmark scripts (#319) 2023-06-30 01:12:59 +04:00