#!/bin/bash

python main.py data=bach model.neomlp_attention.ffn_dim=256 model.embed_dim=64 \
  model.neomlp_attention.num_heads=4 model.neomlp_attention.num_layers=3 \
  data.batch_size=4096 optim.lr=5e-3 model.pos_embedding_dim=512 model.num_nodes=8

python main.py data=bigbuckbunny model.neomlp_attention.ffn_dim=1024 \
  model.embed_dim=512 model.neomlp_attention.num_heads=8 \
  model.neomlp_attention.num_layers=4 data.batch_size=8192 data.eval_batch_size=8192 \
  model.pos_embedding_dim=512 optim.lr=5e-4
