python3 main.py --train_task var-copy --eval_task var-copy --layer1 SSM --layer2 TF --lr 1e-2 --hidden_size 4

jupyter nbconvert --execute --to notebook --inplace run.ipynb &