clarify in readme

speed_benchmark/README.md

 Steps:
 
-1. Run `python speed_benchmark/speed_benchmark.py` which times operations and writes their time to `speed_benchmark/info_a100_py2.jsonl` (feel free to change the name for your profiling).
-2. Run `python speed_benchmark/make_plot_with_jsonl.py`, which produces the `speed_benchmark/plot_with_info.pdf`.
+1. Run `python speed_benchmark/speed_benchmark.py` which times operations and writes their time to `speed_benchmark/info_a100_py2.jsonl` (change the name of the jsonl to a different name for your profiling).
+2. Run `python speed_benchmark/make_plot_with_jsonl.py`, which produces the `speed_benchmark/plot_with_info.pdf`. Again make sure you change the jsonl which is being processed.
\ No newline at end of file

speed_benchmark/make_plot_with_jsonl.py

@@ -20,6 +20,7 @@ if __name__ == '__main__':
 
     ax = fig.add_subplot(gs[0, 0])
 
+    # TODO: change this to what you want.
     rdf = pd.read_json('speed_benchmark/info_a100_py2.jsonl', lines=True)
     df = rdf[rdf.batch_size == batch_size_for_plot1]
 

speed_benchmark/speed_benchmark.py

@@ -97,5 +97,6 @@ if __name__ == '__main__':
 
                 info_json = json.dumps(info)
 
-                with open("speed_benchmark/info_a100_py2.jsonl", "a") as file:
+                # TODO: change this to what you want.
+                with open("speed_benchmark/info.jsonl", "a") as file:
                     file.write(info_json + "\n")