search

ethanfetaya / NRI

Neural relational inference for interacting systems - pytorch
MIT License
732 stars 157 forks source link

Cleanup #1

Closed tkipf closed 6 years ago

tkipf commented 6 years ago

Clean up training files and rename modules to be in line with the description in the paper. This also updates the Readme file with further usage instructions

tkipf commented 6 years ago

@ethanfetaya: can you merge? I don’t have write access, unfortunately.