c9dc55111e8fdf4ab89cdd7976b17b7336bb2770,benchmark.py,,,#,53

Before Change

if __name__ == "__main__":

    experiments = {
        "client_batch_size": [1, 4, 16, 64, 256, 1024, 4096, 8192, 16384],
        "max_batch_size": [32, 64, 128, 256, 512, 1024],
        "max_seq_len": [16, 32, 64, 128, 256, 512],
        "num_client": [1, 2, 4, 8, 16, 32, 64, 128],
        "pooling_layer": [[-j] for j in range(1, 13)]
    }

    fp = open("benchmark-%d-fp16-%s.result" % (common["num_worker"], common["fp16"]), "w")
    for var_name, var_lst in experiments.items():
        // set common args

After Change

        for i, j in zip(var_lst, avg_speed):
            fp.write("|%s|%d|\n" % (i, j))
            fp.flush()
        fp.write("\n")
    fp.close()

Instances