diff --git a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs index e9e1e75f..36b2c0ba 100644 --- a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs +++ b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs @@ -18,7 +18,7 @@ namespace Tensorflow.Benchmark.Leak var modelDir = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location); var ClassifierModelPath = Path.Combine(modelDir, "Leak", "TestModel", "saved_model"); - for (var i = 0; i < 1000; i++) + for (var i = 0; i < 50; i++) { var session = Session.LoadFromSavedModel(ClassifierModelPath);