Browse Source

adding saved model cleanup benchmark

tags/TensorFlowOpLayer
Luke Bollam Esther Hu 4 years ago
parent
commit
4cbf362fc6
5 changed files with 47 additions and 0 deletions
  1. +31
    -0
      src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
  2. BIN
      src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/saved_model.pb
  3. BIN
      src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/variables/variables.data-00000-of-00001
  4. BIN
      src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/variables/variables.index
  5. +16
    -0
      src/TensorFlowNet.Benchmarks/Tensorflow.Benchmark.csproj

+ 31
- 0
src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs View File

@@ -0,0 +1,31 @@
using BenchmarkDotNet.Attributes;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;

namespace Tensorflow.Benchmark.Leak
{
public class SavedModelCleanup
{
[Benchmark]
public void Run()
{
var modelDir = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
var ClassifierModelPath = Path.Combine(modelDir, "Leak", "TestModel", "saved_model");

for (var i = 0; i < 1000; i++)
{
var session = Session.LoadFromSavedModel(ClassifierModelPath);

session.graph.Exit();
session.graph.Dispose();
session.Dispose();
}
}
}
}

BIN
src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/saved_model.pb View File


BIN
src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/variables/variables.data-00000-of-00001 View File


BIN
src/TensorFlowNet.Benchmarks/Leak/TestModel/saved_model/variables/variables.index View File


+ 16
- 0
src/TensorFlowNet.Benchmarks/Tensorflow.Benchmark.csproj View File

@@ -44,4 +44,20 @@
<ProjectReference Include="..\TensorFlowNET.Keras\Tensorflow.Keras.csproj" /> <ProjectReference Include="..\TensorFlowNET.Keras\Tensorflow.Keras.csproj" />
</ItemGroup> </ItemGroup>


<ItemGroup>
<Folder Include="Leak\TestModel\saved_model\assets\" />
</ItemGroup>

<ItemGroup>
<None Update="Leak\TestModel\saved_model\saved_model.pb">
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
</None>
<None Update="Leak\TestModel\saved_model\variables\variables.data-00000-of-00001">
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
</None>
<None Update="Leak\TestModel\saved_model\variables\variables.index">
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
</None>
</ItemGroup>

</Project> </Project>

Loading…
Cancel
Save