search

kimhc6028 / relational-networks

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)
https://arxiv.org/pdf/1706.01427.pdf
BSD 3-Clause "New" or "Revised" License
812 stars 160 forks source link

Magic Number Question #8

Closed talbaumel closed 7 years ago

talbaumel commented 7 years ago

Hi, in line 21 I had a hard time understanding the calculation (24+2)*2+11 The input of the linear layer should be two objects each encoded as a vector size of 24. Where did the +2 and +11 came from?

Thanks!

kimhc6028 commented 7 years ago

Hi. +2 means x,y coordinate of the input object. And +11 is number of question vector. You can see that question vector is concatenated to object vectors at line 74

talbaumel commented 7 years ago

Got it Thanks!