Skip to content

Commit 9667fe9

Browse files
committed
update documentation
1 parent 3baf850 commit 9667fe9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

neurobench/benchmarks/benchmark.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@ def __init__(
5151
Args:
5252
model: A NeuroBenchModel.
5353
dataloader: A PyTorch DataLoader.
54-
preprocessors: A list of NeuroBenchPreProcessors.
55-
postprocessors: A list of NeuroBenchPostProcessors.
56-
metric_list: A list of lists of strings of metrics to run.
57-
First item is static metrics, second item is data metrics.
54+
preprocessors: A list of NeuroBenchPreProcessors or callable functions (e.g. lambda) with matching interfaces.
55+
postprocessors: A list of NeuroBenchPostProcessors or callable functions (e.g. lambda) with matching interfaces.
56+
metric_list: A list of lists of StaticMetric and WorkloadMetric classes of metrics to run.
57+
First item is StaticMetrics, second item is WorkloadMetrics.
5858
"""
5959

6060
self.model = model

0 commit comments

Comments
 (0)