ToruOwO/RNEM-pytorch

A PyTorch implementation of R-NEM

RNEM-pytorch

A PyTorch implementation of the paper "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions"

Training

python main.py [ -u | --usage ] train [...]

Evaluation

python main.py [ -u | --usage ] eval [...]

Rollout

python main.py [ -u | --usage ] rollout [...]