Pass the right arguments to MultiHeadDotProductAttention following commit #27555e8

google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

Apache License 2.0

3.34k stars 440 forks source link

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Pass the right arguments to MultiHeadDotProductAttention following commit #27555e8