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

Fixed mode of native SSD for val

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-10 14:17:50 +02:00
parent 9dbfa5c640
commit 15498549b7
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

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