Skip to content

Commit 4459f71

Browse files
implement export_savedmodel based on tf version
1 parent 4b8b2be commit 4459f71

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

R/model-persistence.R

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,8 +250,17 @@ export_savedmodel.keras.engine.training.Model <- function(object, export_dir_bas
250250

251251
sess <- backend()$get_session()
252252

253-
input_info <- model_to_tensors_info(object$input_layers, "input")
254-
output_info <- model_to_tensors_info(object$output_layers, "output")
253+
if (tensorflow::tf_version() < '1.4') {
254+
input_tensor <- object$input_layers
255+
output_tensor <- object$output_layers
256+
}
257+
else {
258+
input_tensor <- object$layers
259+
output_tensor <- object$layers
260+
}
261+
262+
input_info <- model_to_tensors_info(input_tensor, "input")
263+
output_info <- model_to_tensors_info(output_tensor, "output")
255264

256265
builder <- tensorflow::tf$saved_model$builder$SavedModelBuilder(export_dir_base)
257266
builder$add_meta_graph_and_variables(

0 commit comments

Comments
 (0)