ethanfetaya / NRI

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

Speedup dataset generation #13

Open lromor opened 5 years ago

lromor commented 5 years ago

Also included a small fix on the l2 norm returning negative values (hence introducing a numpy warning when computing the pow).