From 4321c196ab6c6bf796589cee30445105e7ba8990 Mon Sep 17 00:00:00 2001 From: Setepenre Date: Fri, 21 Jun 2024 13:23:38 -0400 Subject: [PATCH] Update main.py --- benchmarks/_template/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/_template/main.py b/benchmarks/_template/main.py index 46fed8e8e..99f0f0adc 100644 --- a/benchmarks/_template/main.py +++ b/benchmarks/_template/main.py @@ -15,7 +15,7 @@ def main(): observer = BenchObserver(batch_size_fn=lambda batch: 1) # optimizer = observer.optimizer(optimizer) - # criterion = observer.optimizer(criterion) + # criterion = observer.criterion(criterion) dataloader = [1, 2, 3, 4]