Skip to content

Conversation

@prakharguptaujjain
Copy link

@prakharguptaujjain prakharguptaujjain commented Mar 31, 2023

For issue #197

The work can be checked out at https://colab.research.google.com/drive/1e9qRYDFh53hZ63dnq0k9rd4QNLvO8vwk?usp=sharing

Screenshots

image

Note-[If the batch_size is set to 4, then the transformer model take approx 4.5GB GPU Ram]

image
ALL TESTS PASSING

@eldraco eldraco self-assigned this May 23, 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.

2 participants