mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +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 |