Speed benchmark #112
Speed benchmark
#112
-
|
This is an amazing project! Thank you! Besides the beatify of einops, I also concern about the running time. |
Beta Was this translation helpful? Give feedback.
Answered by
arogozhnikov
Apr 22, 2021
Replies: 1 comment
-
|
@xvjiarui speed is similar, you can walk around different examples here: Even on small examples speed is similar, and that's without tracing If you have really small tensors, just trace your model and you'll get optimal speed |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
arogozhnikov
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@xvjiarui speed is similar, you can walk around different examples here:
Even on small examples speed is similar, and that's without tracing
https://github.com/arogozhnikov/einops/blob/master/docs/source_examples/Pytorch.ipynb
If you have really small tensors, just trace your model and you'll get optimal speed