Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding a Colab Notebook for ease of use #4

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

sayakpaul
Copy link

This PR adds a Colab Notebook in order to make it easier for people to try out your awesome work easily.

Note that I have just added an explicit device placement in the Grad Rollout code. That's the only change in terms of code. The Colab Notebook mostly follows code from your vit_explain.py script.

If the PR gets merged the Colab Notebook needs to be updated to accommodate for the following:

  • I am currently cloning the forked repository to reflect the device placement code update.
  • Colab Notebook link.

@sayakpaul
Copy link
Author

@jacobgil just checking to see if you had the time to consider this.

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.

1 participant