#!/bin/bash python -m train \ --gpu_id 2 \ --seed 123 \ --logger \ --non_blocking \ --pin_memory \ --batch_size 64 \ --num_workers 16 \ --num_epoch 300 \ --lr 5e-4 \ --dropout 0.1 \ --model_type tom_cm \ --aggr no_tom \ --hidden_dim 64