RIM-MovingMNIST

Use RIMs to do moving MNIST prediction task.

networks.py: - Encoder + RNN + Decoder

rnn_models.py: - RIM, SCOFF

RIM: attention.py + group_operations.py

Name		Name	Last commit message	Last commit date
Latest commit History 488 Commits
.vscode		.vscode
configs		configs
datasets		datasets
logbook		logbook
scripts		scripts
slot_attn		slot_attn
utils		utils
.DS_Store		.DS_Store
.gitignore		.gitignore
README.md		README.md
argument_parser.py		argument_parser.py
ari.py		ari.py
attentions.py		attentions.py
gen_mmnist.py		gen_mmnist.py
group_operations.py		group_operations.py
hyp_sweep.yaml		hyp_sweep.yaml
mmnist_sample.pt		mmnist_sample.pt
networks.py		networks.py
num_module_sweep.yaml		num_module_sweep.yaml
pretrain_autoencoder.py		pretrain_autoencoder.py
rnn_models.py		rnn_models.py
saliency_map.py		saliency_map.py
slot_num_sweep_scoff.yaml		slot_num_sweep_scoff.yaml
test_mmnist.py		test_mmnist.py
train_mmnist.py		train_mmnist.py
visualize_logtensor.py		visualize_logtensor.py

Provide feedback