prateekjoshi565/model_start_bert.py

Created July 17, 2020 17:28

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/prateekjoshi565/e41850d47c00803f66b8c1dfe94f21f0.js"></script>
Save prateekjoshi565/e41850d47c00803f66b8c1dfe94f21f0 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

model_start_bert.py

	# 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