./main.py --task=memory --model=4M-C --nb_epochs=5

./main.py --task=grid --model=37M-C