Use custom loss in benchmarks

MilesCranmer · Apr 24, 2023 · 9418230 · 9418230
1 parent bd0d4ce
commit 9418230
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/benchmark/benchmarks.jl b/benchmark/benchmarks.jl
@@ -23,6 +23,7 @@ function create_search_benchmark()
         maxsize=30,
         verbosity=0,
         progress=false,
+        loss=(pred, target) -> (pred - target)^2,
         extra_kws...,
     )
     seeds = 1:3