A PyTorch implementation of the paper "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions"
python main.py [ -u | --usage ] train [...]
python main.py [ -u | --usage ] eval [...]
python main.py [ -u | --usage ] rollout [...]