はじめに Tensorflowで転移学習させたモデルを保存し、再度それを読み込んで処理した時に諸々問題が発生した。 半日ほどハマったので、備忘録を兼ねてメモ。 問題の状況 1 tensorflow.train.saver.restore()を使って学習済みモデルを読み込んだ。この際に再学習させるモデルは若干変えたため、以下のように一部のパラメータだけを読み込んだ。 sess = tf.Session() ...... all_vars = tf.all_variables() ...... sess.run(tf.global_variables_initializer()) ...... var_to_restore = [] for num, var1 in enumerate(all_vars): _, deter, _ = var1.name.split('/', 2) if de
![Tensorflowで転移学習させたモデルを保存する際の注意点 - Qiita](https://cdn-ak-scissors.b.st-hatena.com/image/square/ea99bf8bda34ba9bbac6a0d6a9c97494535cb74b/height=288;version=1;width=512/https%3A%2F%2Fqiita-user-contents.imgix.net%2Fhttps%253A%252F%252Fcdn.qiita.com%252Fassets%252Fpublic%252Farticle-ogp-background-9f5428127621718a910c8b63951390ad.png%3Fixlib%3Drb-4.0.0%26w%3D1200%26mark64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZ3PTkxNiZoPTMzNiZ0eHQ9VGVuc29yZmxvdyVFMyU4MSVBNyVFOCVCQiVBMiVFNyVBNyVCQiVFNSVBRCVBNiVFNyVCRiU5MiVFMyU4MSU5NSVFMyU4MSU5QiVFMyU4MSU5RiVFMyU4MyVBMiVFMyU4MyU4NyVFMyU4MyVBQiVFMyU4MiU5MiVFNCVCRiU5RCVFNSVBRCU5OCVFMyU4MSU5OSVFMyU4MiU4QiVFOSU5QSU5QiVFMyU4MSVBRSVFNiVCMyVBOCVFNiU4NCU4RiVFNyU4MiVCOSZ0eHQtY29sb3I9JTIzMjEyMTIxJnR4dC1mb250PUhpcmFnaW5vJTIwU2FucyUyMFc2JnR4dC1zaXplPTU2JnR4dC1jbGlwPWVsbGlwc2lzJnR4dC1hbGlnbj1sZWZ0JTJDdG9wJnM9M2Y0MzdjYTAxMWY3MTE1YTQ5ODRjMzBhMDY5NjQ2NjE%26mark-x%3D142%26mark-y%3D112%26blend64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZ3PTYxNiZ0eHQ9JTQwbWFzYXRha2E0NiZ0eHQtY29sb3I9JTIzMjEyMTIxJnR4dC1mb250PUhpcmFnaW5vJTIwU2FucyUyMFc2JnR4dC1zaXplPTM2JnR4dC1hbGlnbj1sZWZ0JTJDdG9wJnM9ZmI2YWE0NzQzMGViZjJhODA0NWJhNGU5ZTRmYzI4Y2Q%26blend-x%3D142%26blend-y%3D491%26blend-mode%3Dnormal%26s%3D2f05f4d1acbbbc5c72a82f3a9f421704)