Output of BertEmbeddings
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 KBKindly help me correct this error.PS: When I tried printing the shape, it was giving me correct output of (batchsize, 512, 768)
2021-05-01T11:22:08Z
The error was basically because of one of my predefined weight tensor was not on cuda. Now everything is working fine. Thanks.