Skip to content

multihead attention version with loop by batch dimension to reduce memory usage#10

Open
krunt wants to merge 4 commits into
learning-at-home:mainfrom
krunt:main
Open

multihead attention version with loop by batch dimension to reduce memory usage#10
krunt wants to merge 4 commits into
learning-at-home:mainfrom
krunt:main

Update tests/test_attn.py

90abdb8
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs