├── .gitattributes ├── README.md ├── dataset.py ├── eval.ipynb ├── model.py └── train.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/.gitattributes -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/dataset.py -------------------------------------------------------------------------------- /eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/eval.ipynb -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/model.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/evelinehong/slot-attention-pytorch/HEAD/train.py --------------------------------------------------------------------------------