2016-06-28 4 views
1

私は、Tensorflow 0.8を使用してディープニューラルネットワークをトレーニングしています。現在、私は2つの全く同じニューラルネットワークN1とN2を定義し、トレーニングループ中にN1をトレーニングし、4回の繰り返しごとに更新された重みをN1からN2にコピーするという問題に直面する。実際には、tf.train.saver.save()を使用して、Disk上の.ckptファイルにすべてのN1ウェイトを保存し、tf.train.saver.restore()を使用してコピー機能と同等の.ckptファイルからそれらのウェイトをロードする方法があることを知っています。しかし、このロード/リロードはトレーニングのスピードに影響を与え、コピーを行う他のより効率的な方法があるのだろうかと思います(たとえば、メモリ内コピーなど)。ありがとう!Tensorflowコピーウェイト問題

+0

ここで私の質問を見てください:https://stackoverflow.com/questions/48547688/tensorflow-keras-copy-weights-from-one-model-to-another?noredirect=1#comment84092982_48547688 – avejidah

答えて

0

あなたのコード/ここでより詳細な情報があれば有益でしょう。ただし、N2を訓練するために使用しているセッションを戻し、N2を訓練したいときにアクセスすることができます。

関連する問題