Loading SavedModel is a lot slower than loading a tf.train.Saver checkpoint

后端 未结 2 1277
陌清茗
陌清茗 2021-02-01 03:40

I changed from tf.train.Saver to the SavedModel format which surprisingly means loading my model from disk is a lot slower (instead of a couple of seconds it takes

2条回答
  •  情书的邮戳
    2021-02-01 04:04

    what can I do to load the model faster?

    Switch back to tf.train.Saver, as your question shows no motivations for using SavedModelBuilder, and makes it clear that elapsed time matters to you. Alternatively, an MCVE that reproduced the timing issue would allow others to collaborate with you on profiling, diagnosing, and fixing any perceived performance issue.

提交回复
热议问题