2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Moved loss metric assignment into begin of epoch

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-11 11:34:04 +02:00
parent f6f6a88c50
commit 83b9e670f8
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/twomartens/masterthesis/ssd.py
View File

@ -458,16 +458,13 @@ def _train_epochs(nr_batches_per_epoch: int,
epoch = 0 epoch = 0
batch_counter = 0 batch_counter = 0
epoch_var.assign(epoch) epoch_var.assign(epoch)
epoch_start_time = None
# define loss variables
ssd_loss_avg = tfe.metrics.Mean(name='ssd_loss', dtype=tf.float32)
# go through data set # go through data set
for x, y in dataset: for x, y in dataset:
if batch_counter == 0: if batch_counter == 0:
# epoch starts # epoch starts
epoch_start_time = time.time() epoch_start_time = time.time()
ssd_loss_avg = tfe.metrics.Mean(name='ssd_loss', dtype=tf.float32)
labels = [] labels = []
for i in range(y.shape[0]): for i in range(y.shape[0]):