python spriteworld_scm_discovery.py --num_epochs 125 --num_runs 1 --num_sprites 4 --batch_size 100 --num_examples 50000 --mask_reg 0. --weight_reg 0. --attn_reg 0. --weight_decay 1e-5 --model_type MMN --results_dir ./mmn_model2