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

Changed mode to training for DropoutSSD to allow partitioning of observations

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-21 11:31:24 +02:00
parent 0056f28e79
commit af9cbc60d2
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

@ -141,7 +141,7 @@ def predict(dataset: tf.data.Dataset,
checkpointables = {} checkpointables = {}
if use_dropout: if use_dropout:
checkpointables.update({ checkpointables.update({
'ssd': DropoutSSD(mode='inference_fast', weights_path=weights_path) 'ssd': DropoutSSD(mode='training', weights_path=weights_path)
}) })
else: else:
checkpointables.update({ checkpointables.update({