FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
TensorFlow.NET/src/TensorFlowNET.Console/MemoryMonitor.cs at master · sbiaudet/TensorFlow.NET · GitHub
sbiaudet
TensorFlow.NET
Repository navigation
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
TensorFlow.NET
/
src
/
TensorFlowNET.Console
/
MemoryMonitor.cs
Copy path
More file actions
More file actions
Latest commit
History
History
History
68 lines (58 loc) · 2.2 KB
Breadcrumbs
TensorFlow.NET
/
src
/
TensorFlowNET.Console
/
MemoryMonitor.cs
Copy path
File metadata and controls
68 lines (58 loc) · 2.2 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
using
System
;
using
System
.
Diagnostics
;
using
System
.
Threading
;
using
System
.
Threading
.
Tasks
;
using
NumSharp
;
using
static
Tensorflow
.
Binding
;
using
static
Tensorflow
.
KerasApi
;
namespace
Tensorflow
{
public
class
MemoryMonitor
{
public
void
WarmUp
(
)
{
TensorShape
shape
=
(
1
,
32
,
32
,
3
)
;
np
.
arange
(
shape
.
size
)
.
astype
(
np
.
float32
)
.
reshape
(
shape
.
dims
)
;
print
(
$
"tensorflow native version: v
{
tf
.
VERSION
}
"
)
;
tf
.
Context
.
ensure_initialized
(
)
;
var
a
=
tf
.
constant
(
np
.
ones
(
10
,
10
)
)
;
var
b
=
tf
.
Variable
(
a
)
;
var
c
=
tf
.
Variable
(
b
)
;
var
d
=
b
*
c
;
print
(
d
.
numpy
(
)
)
;
GC
.
Collect
(
)
;
GC
.
WaitForPendingFinalizers
(
)
;
}
public
void
Execute
(
int
epoch
,
int
iterate
,
Action
<
int
,
int
>
process
)
{
GC
.
Collect
(
)
;
GC
.
WaitForPendingFinalizers
(
)
;
var
initialTotalMemory
=
Process
.
GetCurrentProcess
(
)
.
PrivateMemorySize64
;
print
(
$
"
{
process
.
Method
.
Name
}
started..."
)
;
for
(
int
i
=
0
;
i
<
epoch
;
i
++
)
{
var
initialMemory
=
Process
.
GetCurrentProcess
(
)
.
PrivateMemorySize64
;
for
(
int
j
=
0
;
j
<
iterate
;
j
++
)
process
(
i
,
j
)
;
keras
.
backend
.
clear_session
(
)
;
GC
.
Collect
(
)
;
GC
.
WaitForPendingFinalizers
(
)
;
var
finalMemory
=
Process
.
GetCurrentProcess
(
)
.
PrivateMemorySize64
;
print
(
$
"Epoch
{
i
}
:
{
Format
(
finalMemory
-
initialMemory
)
}
."
)
;
}
var
finalTotalMemory
=
Process
.
GetCurrentProcess
(
)
.
PrivateMemorySize64
;
print
(
$
"Memory usage difference:
{
Format
(
finalTotalMemory
-
initialTotalMemory
)
}
/
{
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"
;
}
}
}
Back
|
FazBrowse Home
|
New Git URL