diff --git a/CHANGELOG.md b/CHANGELOG.md
index 15090708aa02..d7b3d0e8d925 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -14,7 +14,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 - Added `predict()` support to the `LightningNodeData` module ([#4884](https://github.com/pyg-team/pytorch_geometric/pull/4884))
 - Added `time_attr` argument to `LinkNeighborLoader` ([#4877](https://github.com/pyg-team/pytorch_geometric/pull/4877), [#4908](https://github.com/pyg-team/pytorch_geometric/pull/4908))
 - Added a `filter_per_worker` argument to data loaders to allow filtering of data within sub-processes ([#4873](https://github.com/pyg-team/pytorch_geometric/pull/4873))
-- Added a `NeighborLoader` benchmark script ([#4815](https://github.com/pyg-team/pytorch_geometric/pull/4815))
+- Added a `NeighborLoader` benchmark script ([#4815](https://github.com/pyg-team/pytorch_geometric/pull/4815), [#4862](https://github.com/pyg-team/pytorch_geometric/pull/4862/files))
 - Added support for `FeatureStore` and `GraphStore` in `NeighborLoader` ([#4817](https://github.com/pyg-team/pytorch_geometric/pull/4817), [#4851](https://github.com/pyg-team/pytorch_geometric/pull/4851), [#4854](https://github.com/pyg-team/pytorch_geometric/pull/4854), [#4856](https://github.com/pyg-team/pytorch_geometric/pull/4856), [#4857](https://github.com/pyg-team/pytorch_geometric/pull/4857), [#4882](https://github.com/pyg-team/pytorch_geometric/pull/4882), [#4883](https://github.com/pyg-team/pytorch_geometric/pull/4883), [#4929](https://github.com/pyg-team/pytorch_geometric/pull/4929), [#4992](https://github.com/pyg-team/pytorch_geometric/pull/4922))
 - Added a `normalize` parameter to `dense_diff_pool` ([#4847](https://github.com/pyg-team/pytorch_geometric/pull/4847))
 - Added `size=None` explanation to jittable `MessagePassing` modules in the documentation ([#4850](https://github.com/pyg-team/pytorch_geometric/pull/4850))
diff --git a/benchmark/loader/neighbor_loader.py b/benchmark/loader/neighbor_loader.py
index cd8f7ed0186b..32e91ffc4ced 100644
--- a/benchmark/loader/neighbor_loader.py
+++ b/benchmark/loader/neighbor_loader.py
@@ -1,4 +1,5 @@
 import argparse
+import ast
 import os.path as osp
 from timeit import default_timer
 
@@ -83,10 +84,14 @@ def run(args: argparse.ArgumentParser) -> None:
     add('--device', default='cpu')
     add('--datasets', nargs="+", default=['arxiv', 'products', 'mag'])
     add('--root', default='../../data')
-    add('--batch-sizes', default=[8192, 4096, 2048, 1024, 512])
-    add('--eval-batch-sizes', default=[16384, 8192, 4096, 2048, 1024, 512])
-    add('--homo-neighbor_sizes', default=[[10, 5], [15, 10, 5], [20, 15, 10]])
-    add('--hetero-neighbor_sizes', default=[[5], [10], [10, 5]], type=int)
+    add('--batch-sizes', default=[8192, 4096, 2048, 1024, 512],
+        type=ast.literal_eval)
+    add('--eval-batch-sizes', default=[16384, 8192, 4096, 2048, 1024, 512],
+        type=ast.literal_eval)
+    add('--homo-neighbor_sizes', default=[[10, 5], [15, 10, 5], [20, 15, 10]],
+        type=ast.literal_eval)
+    add('--hetero-neighbor_sizes', default=[[5], [10], [10, 5]],
+        type=ast.literal_eval)
     add('--num-workers', default=0)
     add('--runs', default=3)