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.

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)