diff --git a/runs/exp02_one_layer/evaluate.py b/runs/exp02_one_layer/evaluate.py index e812974..37dccf9 100644 --- a/runs/exp02_one_layer/evaluate.py +++ b/runs/exp02_one_layer/evaluate.py @@ -1,4 +1,3 @@ -import os from pathlib import Path import matplotlib.pyplot as plt @@ -94,7 +93,8 @@ def load_checkpoint_and_evaluate(test_loader: DataLoader, checkpoint_path: Path, from config import CONFIG from dataset import XRR1LayerDataset - exp_dir = Path(os.environ["DATA"], r"XRR_AI\data\one_layer\run") + # The run main.py trains, not another experiment's folder. + exp_dir = CONFIG["base_dir"] / CONFIG["exp_name"] h5_file = exp_dir / "dataset.h5" stats_file = exp_dir / "stats.pt" checkpoint_file = exp_dir / "best.pt" diff --git a/runs/exp02_one_layer/evaluate_with_real_data.py b/runs/exp02_one_layer/evaluate_with_real_data.py index 663eb07..498573d 100644 --- a/runs/exp02_one_layer/evaluate_with_real_data.py +++ b/runs/exp02_one_layer/evaluate_with_real_data.py @@ -176,11 +176,13 @@ def evaluate_with_plots(model, data_loader: DataLoader, stats_path: Path, out_pr # Example usage block (사용자 환경에 맞게 수정) # ------------------------------ if __name__ == "__main__": + from config import CONFIG from model import XRR1DRegressor # ----------------------------- # 경로 설정 # ----------------------------- - exp_dir = Path(os.environ["DATA"], r"XRR_AI\data\one_layer\run") + # The run main.py trains, not another experiment's folder. + exp_dir = CONFIG["base_dir"] / CONFIG["exp_name"] stats_file = exp_dir / "stats.pt" checkpoint_file = exp_dir / "best.pt" out_dir = exp_dir / "eval_real" diff --git a/runs/exp04_one_genx/evaluate.py b/runs/exp04_one_genx/evaluate.py index 39b4630..09979d8 100644 --- a/runs/exp04_one_genx/evaluate.py +++ b/runs/exp04_one_genx/evaluate.py @@ -1,4 +1,3 @@ -import os from pathlib import Path import matplotlib.pyplot as plt @@ -94,7 +93,8 @@ def load_checkpoint_and_evaluate(test_loader: DataLoader, checkpoint_path: Path, from config import CONFIG from dataset import XRR1LayerDataset - exp_dir = Path(os.environ["DATA"], r"XRR_AI\data\one_layer\run") + # The run main.py trains, not another experiment's folder. + exp_dir = CONFIG["base_dir"] / CONFIG["exp_name"] h5_file = exp_dir / "dataset.h5" stats_file = exp_dir / "stats.pt" checkpoint_file = exp_dir / "best.pt" diff --git a/runs/exp04_one_genx/evaluate_with_real_data.py b/runs/exp04_one_genx/evaluate_with_real_data.py index 663eb07..498573d 100644 --- a/runs/exp04_one_genx/evaluate_with_real_data.py +++ b/runs/exp04_one_genx/evaluate_with_real_data.py @@ -176,11 +176,13 @@ def evaluate_with_plots(model, data_loader: DataLoader, stats_path: Path, out_pr # Example usage block (사용자 환경에 맞게 수정) # ------------------------------ if __name__ == "__main__": + from config import CONFIG from model import XRR1DRegressor # ----------------------------- # 경로 설정 # ----------------------------- - exp_dir = Path(os.environ["DATA"], r"XRR_AI\data\one_layer\run") + # The run main.py trains, not another experiment's folder. + exp_dir = CONFIG["base_dir"] / CONFIG["exp_name"] stats_file = exp_dir / "stats.pt" checkpoint_file = exp_dir / "best.pt" out_dir = exp_dir / "eval_real"