TensorFlow.NET/src/TensorFlowNET.Console/MemoryMonitor.cs at master · pathway27/TensorFlow.NET · GitHub

History

51 lines (44 loc) · 1.51 KB

Raw

using System;

using System.Collections.Generic;

using System.Diagnostics;

using System.Text;

using static Tensorflow.Binding;

namespace Tensorflow

{

public class MemoryMonitor

{

public void WarmUp()

{

print(tf.VERSION);

}

public void Execute(int epoch, int iterate, Action<int> process)

{

/*GC.Collect();

GC.WaitForPendingFinalizers();

GC.Collect();*/

print($"{process.Method.Name} started...");

for (int i = 0; i < epoch; i++)

{

var initialMemory = Process.GetCurrentProcess().PrivateMemorySize64;// GC.GetTotalMemory(true);

process(iterate);

var finalMemory = Process.GetCurrentProcess().PrivateMemorySize64; //GC.GetTotalMemory(true);

print($"Epoch {i}: {Format(finalMemory - initialMemory)}.");

}

GC.Collect();

GC.WaitForPendingFinalizers();

GC.Collect();

print($"Total {process.Method.Name} usage {Format(Process.GetCurrentProcess().PrivateMemorySize64)}");

}

private string Format(long usage)

{

if (usage < 0)

return $"-{Format(0 - usage)}";

if (usage <= 1024 && usage >= 0)

return $"{usage} Bytes";

else if (usage > 1024 && usage <= 1024 * 1024)

return $"{usage / 1024} KB";

else

return $"{usage / 1024 / 1024} MB";

}

TensorFlow.NET/src/TensorFlowNET.Console/MemoryMonitor.cs at master · pathway27/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

MemoryMonitor.cs

Latest commit

History

Breadcrumbs

MemoryMonitor.cs

File metadata and controls