Tensorflow, restore variables in a specific device

前端 未结 2 2003
萌比男神i
萌比男神i 2021-01-14 03:33

Maybe my question is a bit naive, but I really didn\'t find anything in the tensorflow documentation.

I have a trained tensorflow model where the variables of it was

2条回答
  •  无人及你
    2021-01-14 04:00

    Use clear_devices flag, ie

    saver = tf.train.import_meta_graph("/tmp/graph.meta", clear_devices=True)
    

提交回复
热议问题