diff --git a/baselines/HimNet/METR-LA.py b/baselines/HimNet/METR-LA.py
index cfe45489..1faf879a 100644
--- a/baselines/HimNet/METR-LA.py
+++ b/baselines/HimNet/METR-LA.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/baselines/HimNet/PEMS-BAY.py b/baselines/HimNet/PEMS-BAY.py
index a9ae50e4..d014c638 100644
--- a/baselines/HimNet/PEMS-BAY.py
+++ b/baselines/HimNet/PEMS-BAY.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/baselines/HimNet/PEMS03.py b/baselines/HimNet/PEMS03.py
index 4af9b6bc..bf61819d 100644
--- a/baselines/HimNet/PEMS03.py
+++ b/baselines/HimNet/PEMS03.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/baselines/HimNet/PEMS04.py b/baselines/HimNet/PEMS04.py
index 5289f7b3..9a53ba04 100644
--- a/baselines/HimNet/PEMS04.py
+++ b/baselines/HimNet/PEMS04.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/baselines/HimNet/PEMS07.py b/baselines/HimNet/PEMS07.py
index b9f84705..b6351905 100644
--- a/baselines/HimNet/PEMS07.py
+++ b/baselines/HimNet/PEMS07.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/baselines/HimNet/PEMS08.py b/baselines/HimNet/PEMS08.py
index 5adb0c12..b50f6beb 100644
--- a/baselines/HimNet/PEMS08.py
+++ b/baselines/HimNet/PEMS08.py
@@ -59,7 +59,7 @@
 CFG = EasyDict()
 # General settings
 CFG.DESCRIPTION = 'An Example Config'
-CFG.GPU_NUM = 8 # Number of GPUs to use (0 for CPU mode)
+CFG.GPU_NUM = 1 # Number of GPUs to use (0 for CPU mode)
 # Runner
 CFG.RUNNER = HimNetRunner
 
@@ -96,7 +96,7 @@
 CFG.MODEL.PARAM = MODEL_PARAM
 CFG.MODEL.FORWARD_FEATURES = [0, 1, 2]
 CFG.MODEL.TARGET_FEATURES = [0]
-CFG.MODEL.SETUP_GRAPH = True
+CFG.MODEL.SETUP_GRAPH = False
 
 ############################## Metrics Configuration ##############################
 
diff --git a/basicts/runners/base_epoch_runner.py b/basicts/runners/base_epoch_runner.py
index 2cb17e92..a774263c 100644
--- a/basicts/runners/base_epoch_runner.py
+++ b/basicts/runners/base_epoch_runner.py
@@ -596,7 +596,7 @@ def inference_pipeline(self, cfg: Optional[Dict] = None, input_data: Union[str,
         result = self.inference(save_result_path=output_data_file_path)
 
         inference_end_time = time.time()
-        self.update_epoch_meter('inference/time', 'inference', inference_end_time - inference_start_time)
+        self.update_epoch_meter('inference/time', inference_end_time - inference_start_time)
 
         self.print_epoch_meters('inference')
 
diff --git a/basicts/runners/base_tsf_runner.py b/basicts/runners/base_tsf_runner.py
index 807f387b..4ee106cb 100644
--- a/basicts/runners/base_tsf_runner.py
+++ b/basicts/runners/base_tsf_runner.py
@@ -131,6 +131,8 @@ def setup_graph(self, cfg: Dict, train: bool):
 
         dataloader = self.build_test_data_loader(cfg=cfg) if not train else self.build_train_data_loader(cfg=cfg)
         data = next(iter(dataloader))  # get the first batch
+        if not train:
+            self.model.eval()
         self.forward(data=data, epoch=1, iter_num=0, train=train)
 
     def count_parameters(self):