sh distributed_train.sh 4 data/random_sr3_10_100 \
 --dataset ckt --model dgdagrnn -b 128 \
 --sched step --epochs 100 --decay-epochs 2.4 --decay-rate .97 \
 --warmup-lr 1e-6 --weight-decay 1e-5 --lr .016 \
 --opt rmsproptf --opt-eps .001 -j 8 --amp 