Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# pass the pre-trained BERT to our define architecture
model = BERT_Arch(bert)
# push the model to GPU
model = model.to(device)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment