import subprocess

models = ['Transformer', 'Reformer', 'Routing', 'Clusterformer']
mls = [50] + [i*100 for i in range(1, 26)]

for max_length in mls:
    print(max_length)
    for model in models:
        subprocess.call(
            'python time_test.py \
            --dataName 20NEWS --modelName %s --max_length %d' % (model, max_length), shell=True)
    print()
    print()