@Kushagra_Bhatia Glad, it worked. Now, onto the next one. Putting the two errors in context, it can be seen something’s off with the weights (self.wb and self.we) now and not hq. Try changing the types of the weights. I am sure that will take care of it. Let us know how it goes.