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
811 stars 162 forks source link

Suggestion #17

Open rosenfeldamir opened 6 years ago

rosenfeldamir commented 6 years ago

Very nice work! I suggest saving the pickle file for the dataset with protocol=2. This is faster in loading and saving and also uses 1.4GB instead of 5GB.