mirror of
https://github.com/bigscience-workshop/petals
synced 2024-11-10 01:10:33 +00:00
10c72acdf4
The previous code was incorrect for the case of `warmup_steps != 1` (this mode was never used, but can be used in future). |
||
---|---|---|
.. | ||
benchmark_forward.py | ||
benchmark_inference.py | ||
benchmark_training.py |