Fix small bugs

Author: deeperlearner

mains/main.py

@@ -10,7 +10,7 @@
 from logger import get_logger
 from parse_config import ConfigParser
 from mains import train, train_mp, test
-from utils import msg_box, consuming_time
+from utils import msg_box, consuming_time, get_by_path
 
 
 if __name__ == "__main__":

scripts/examples.sh

@@ -36,11 +36,11 @@ while getopts "abc" flag; do
       RUN_ID=${VERSION}
 
       # use optuna to find the best h.p.
-      # python3 mains/main.py --optuna --mp -c "configs/$CONFIG.json" --mode train \
-      #     --run_id $RUN_ID --name $EXP
-      # python3 mains/main.py -c "saved/$EXP/$RUN_ID/best_hp/${CONFIG##*/}.json" --mode test \
-      #     --resume "saved/$EXP/$RUN_ID/best_hp/fold_0_model_best.pth" --run_id $RUN_ID --k_fold 1 --bootstrapping
-      # exit
+      python3 mains/main.py --optuna --mp -c "configs/$CONFIG.json" --mode train \
+          --run_id $RUN_ID --name $EXP
+      python3 mains/main.py -c "saved/$EXP/$RUN_ID/best_hp/${CONFIG##*/}.json" --mode test \
+          --resume "saved/$EXP/$RUN_ID/best_hp/fold_0_model_best.pth" --run_id $RUN_ID --k_fold 1 --bootstrapping
+      exit
 
       # given h.p. with k_fold = 1
       python3 mains/main.py -c "configs/$CONFIG.json" --mode train \

tune/objective.py

@@ -1,7 +1,7 @@
 from logger import get_logger
 from mains import train, train_mp
 from parse_config import ConfigParser
-from utils import msg_box, consuming_time
+from utils import msg_box, consuming_time, get_by_path
 
 objective_results = []
 
@@ -17,7 +17,8 @@ def objective(trial):
     }
     config = ConfigParser(modification)
     logger = get_logger("optuna")
-    max_min, mnt_metric = config["trainer"]["kwargs"]["monitor"].split()
+    keys = ["trainers", "trainer", "kwargs", "monitor"]
+    max_min, mnt_metric = get_by_path(config, keys).split()
     k_fold = config["cross_validation"]["k_fold"]
     msg = msg_box("Optuna progress")
     i, N = len(objective_results), config["optuna"]["n_trials"]