minor changes for pytorch_v1.0

ethanfetaya / NRI

Neural relational inference for interacting systems - pytorch

MIT License

739 stars 157 forks source link

Open hengfun opened 5 years ago

hengfun commented 5 years ago

Minor changes to get train.py running for pytorch v1.0

tkipf commented 5 years ago

Looks good to me, thanks! I suggest we leave this PR open (for reference) until we decide to officially support PyTorch 1.0.