簡體   English   中英

如何與其他用戶共享張量流權重

[英]How to share tensorflow weights with other users

是否可以與其他用戶共享tensorflow檢查點文件(平台和CPU / GPU獨立)? 我分享了DeconvNet的Tensorflow實現,現在我想提供訓練有素的權重。 我可以簡單地上傳保存的模型,還是有另一種tf方法? 我問是因為我讀了一個教程,稱重是使用numpy.savetxt存儲的,然后在numpy.savetxt重量時恢復了。 但是此方法用於使用非常小的網絡的MNIST示例。

謝謝!

你可以保存元圖+提供代碼,以恢復和運行模式- http://tensorflow.org/how_tos/meta_graph

缺點之一是它不提供要進給/獲取的張量的注釋,因此您需要提供一些代碼以顯示如何使用它。

SavedModel是TensorFlow檢查點格式的下一個迭代, SavedModel此問題,但目前尚無太多文檔。

在二進制模式下使用pickle來轉儲和加載大型numpy矩陣,並且效果很好。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM