python -m train_Simulation \
       -data-dir ../dataset \
       -T 60 \
       -b 2048 \
       -epochs 200 \
       -device cuda:1 \
       -net_name SNN \
       -lr 0.001 \
       -tau 2.0 \
       -opt 'adam' \