Skip to content

Conversation

@a-parida12
Copy link
Contributor

Fixes #6032 .

Description

specified the data type of the att_matrix to be compliant with torch.jit.compile requirements for conditionals

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • Breaking change (fix or new feature that would cause existing functionality to change).
  • New tests added to cover the changes.
  • Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
  • Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
  • In-line docstrings updated.
  • Documentation updated, tested make html command in the docs/ folder.

@wyli
Copy link
Contributor

wyli commented Apr 6, 2023

/black

wyli added 2 commits April 11, 2023 14:19
Signed-off-by: Wenqi Li <wenqil@nvidia.com>
…tten-matrix

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
@wyli
Copy link
Contributor

wyli commented Apr 11, 2023

/build

Copy link
Contributor

@wyli wyli left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks, all pre-merge tests work fine, I'm merging this for more tests.

@wyli wyli merged commit d14e8f0 into Project-MONAI:dev Apr 11, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Access self-attention matrix of vision transformer

2 participants