remove redundant files

Author: liecn

.gitignore

@@ -213,4 +213,6 @@ log_0824
 log_0828
 deprecated
 data_test
-run.sh
+run.sh
+run_c.sh
+run_parsimon_eval.sh

README.md

@@ -80,7 +80,6 @@ python gen_ckpt.py --dir_output=/data1/lichenni/m3/ckpts
 Note the checkpoints will be saved in the `ckpts` directory, one is for the Llama-2 model and the other is for the 2-layer MLP model.
 
 7. To replicate paper results in Section 5.2, run the following in the `parsimon-eval/expts/fig_8` directory:
-Note all commands can be found in `parsimon-eval/run.sh`
 
 ```bash
 cargo run --release -- --root=./data --mixes spec/all_dctcp.mix.json ns3-config

clibs/run.sh

@@ -34,9 +34,7 @@
 # set parameters
 model_trained_dir=f"{args.dir_output}/m3_shard2000_nflows1_nhosts3_nsamples20_lr10Gbps/version_0"
 output_dir=f"./ckpts"
-model_id="_config_e421"
-# model_id="_e466"
-# model_id="_hpcc_e447"
+model_id=""
 class m3_inference:
     def __init__(self):
         self.bucket_thold = 1
@@ -53,7 +51,7 @@ def __init__(self):
             training_config = config["training"]
         n_params=dataset_config["n_params"]
         model = FlowSimTransformer_Path.load_from_checkpoint(
-            f"{self.dir_train}/checkpoints/best{model_id}.ckpt",
+            f"{self.dir_train}/checkpoints/last{model_id}.ckpt",
             map_location=DEVICE,
             n_layer=model_config["n_layer"],
             n_head=model_config["n_head"],