dataset is included in the repo to make it easy command line run python3 run.py --config ./train_configs/mnist/train_roast_tloop.py change the merge strategy to merge-wt to see the successful experiment.