File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments