Output of BertEmbeddings

Models
1

I want to do some changes in the BertModel class based upon my use case. As such I wanted to look at some of the outputs in general for better understanding.
When I tried printing the BertEmbeddings output inside BertModel, it is giving me an illegal memory access RuntimeError. I have attached the screenshot below.

image
image991×258 15.7 KB

Kindly help me correct this error.
PS: When I tried printing the shape, it was giving me correct output of (batchsize, 512, 768)

2

The error was basically because of one of my predefined weight tensor was not on cuda. Now everything is working fine. Thanks.