Commit b68706d5 authored by Alexander Fuchs's avatar Alexander Fuchs
Browse files

Removed redundancies, fixed summary bug

parent 5441bc86
......@@ -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"):
......
......@@ -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,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment