ethanfetaya / NRI

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

Fix missing argument error #3

Closed tkipf closed 6 years ago

tkipf commented 6 years ago

This fixes https://github.com/ethanfetaya/NRI/issues/2

tkipf commented 6 years ago

@ethanfetaya Can you merge? Alternatively feel free to give me write access ;-)