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

Changed SSD to training mode to test origin of memory leak

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-01 15:07:43 +02:00
parent 70df855304
commit 280dbbc878
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -144,7 +144,7 @@ def predict(dataset: tf.data.Dataset,
}) })
else: else:
checkpointables.update({ checkpointables.update({
'ssd': SSD(mode='inference_fast', weights_path=weights_path) 'ssd': SSD(mode='training', weights_path=weights_path)
}) })
if checkpoint_path is not None: if checkpoint_path is not None: