Implementation of Relation Network and Recurrent Relational Network using PyTorch v1.3. Original papers: (RN) https://arxiv.org/abs/1706.01427 (RRN): https://arxiv.org/abs/1711.08028
MIT License
19
stars
7
forks
source link
Adds necessary code to pass dict_size automatically #8
Most important stuff is the lambda function that passes the dict_size to the batchify function in the collate function on train.py. The rest is just to add the parameter to pass the value.
Most important stuff is the lambda function that passes the dict_size to the batchify function in the collate function on train.py. The rest is just to add the parameter to pass the value.