rokkiの備忘録

地道こそ近道

tensorflow

今日学んだこと#0011

tensorflow上でモデルの重みの保存と読み込みに,思いの外手間取った. 結論 tf.train.Saver()に重みのリストを渡すと,保存または読み込む.何も渡さないと,すべて保存しようとする. よって,事前に学習した重みを読み込むときは,variableのリストを渡し…

今日学んだこと#0010

tensroflowのAPI tensorflow実装のモデルをpytorchで実装するために,必要なtensorflowのAPIをまとめていく. 主に,下記サイトを参考にした. dev.classmethod.jp 型取得:get_shape() >>> import tensorflow as tf >>> tensor = tf.constant([[1, 1], [1, …