Skip to content
GitLab
Menu
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
Alexander Fuchs
tensorflow2_trainer_template
Commits
b68706d5
Commit
b68706d5
authored
Aug 12, 2020
by
Alexander Fuchs
Browse files
Removed redundancies, fixed summary bug
parent
5441bc86
Changes
2
Hide whitespace changes
Inline
Side-by-side
src/utils/summary_utils.py
View file @
b68706d5
...
...
@@ -45,7 +45,7 @@ class Summaries(object):
for
mode
in
self
.
modes
:
for
key
in
self
.
scalar_summary_names
:
self
.
scalar_summaries
[
mode
+
'_'
+
key
]
=
tf
.
keras
.
metrics
.
Mean
(
'
mode_'
+
key
,
dtype
=
tf
.
float32
)
self
.
scalar_summaries
[
mode
+
'_'
+
key
]
=
tf
.
keras
.
metrics
.
Mean
(
mode
+
'
_'
+
key
,
dtype
=
tf
.
float32
)
def
update
(
self
,
scalars
,
mode
=
"train"
):
...
...
src/utils/trainer.py
View file @
b68706d5
...
...
@@ -18,10 +18,6 @@ class ModelTrainer():
epochs
,
eval_fns
,
model_settings
=
[],
optimizer_types
=
[],
base_learning_rates
=
[],
learning_rate_fns
=
[],
init_data
=
[],
summaries
=
None
,
start_epoch
=
0
,
num_train_batches
=
None
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment