Skip to content
Snippets Groups Projects

Draft: Release 5.0

Open Cresson Remi requested to merge 57-update_tensorflow_version into develop
2 files
+ 2
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -90,7 +90,7 @@ def create_dataset(
tf_ds = dataset.get_tf_dataset(
batch_size=batch_size,
preprocessing_fn=fcnn_model.dataset_preprocessing_fn,
targets_keys=targets_keys or [fcnn_model.TARGET_NAME]
targets_keys=targets_keys or [fcnn_model.OUTPUT_SOFTMAX_NAME]
)
return tf_ds
Loading