From 4915806442c31a0c1ed658156f197ded3c57ba6b Mon Sep 17 00:00:00 2001 From: Thomas Kipf Date: Sun, 28 Oct 2018 19:10:34 +0100 Subject: [PATCH] Update README.md --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 7560407..09910d1 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ This repository contains the official PyTorch implementation of: **Neural relational inference for interacting systems.** Thomas Kipf*, Ethan Fetaya*, Kuan-Chieh Wang, Max Welling, Richard Zemel. -https://arxiv.org/abs/1802.04687 (*: equal contribution)˚ +https://arxiv.org/abs/1802.04687 (*: equal contribution) ![Neural Relational Inference (NRI)](nri.png) @@ -24,6 +24,8 @@ python generate_dataset.py ``` This generates the springs dataset, use `--simulation charged` for charged particles. +Note: Make sure to use the same preprocessing and evaluation scripts (check the loss function as well) as in our code release to get comparable results. + ### Run experiments From the project's root folder, simply run