我熟悉tf.train.Saver,它允许我将变量保存到磁盘上。
但是,有没有可能直接将它们获取为字符串?或者将它们写入一个打开的流中?类似于这样:
binary_string_rep = saver.save(sess)db.save(binary_string_rep)
还是我必须先写入磁盘,然后再读回来?
回答:
您可以将任何张量转换为Numpy数组。然后按您的意愿使用Numpy数组!例如,您的模型中有一个名为W_layer1
的第一层权重。您可以将其转换为Numpy数组,如下所示:W_numpy = W_layer1.eval()
,这也在这里提到了:将张量转换为Numpy数组