問題タブ [tensorflow-hub]

tensorflow を使用して GAN をトレーニングし、ジェネレーターとディスクリミネーターを tensorflow_hub モジュールとしてエクスポートしたいと考えています。
そのために:
- GAN アーキテクチャを tensorflow で定義します
- トレーニングしてチェックポイントを保存します-
次のようなさまざまなタグで module_spec を作成します:
(set(), {'batch_size': 8, 'model': 'gen'})
({'bs8', 'gen'}, {'batch_size': 8, 'model': 'gen'})
({'bs8', 'disc'}, {'batch_size': 8, 'model': 'disc'})
- トレーニング中に保存した checkpoint_path を使用して、tf_hub_path で module_spec でエクスポートします

次に、次のコマンドでジェネレーターをロードできます。

しかし、同様のコマンドを使用してディスクリミネーターをロードしようとすると:

エラーが発生しました：

したがって、ディスクリミネーターに存在する変数は、ディスク上のモジュールに保存されていないと結論付けました。私が想像したさまざまなエラーの原因を確認しました。

• モジュール仕様が正しく定義されていること。そのために、モデルをトレーニングし、モジュール スペックを作成し、その module_spec からモジュールを直接ロードすることにしました。これは、ジェネレーターとディスクリミネーターでうまく機能しました。次に、私のmodule_specが正しいと仮定しました

• 次に、チェックポイントがすべての変数をグラフに正しく保存しているかどうか疑問に思っていました。

  したがって、すべての変数がチェックポイント内に正しく保存されていることがわかりました。ジェネレーターに関連する 2 つの変数のみが、ディスク上の tf ハブ モジュールに正しくエクスポートされました。

最後に、私のエラーは次のものから来ていると思います：

checkpoint_path から変数をエクスポートするために、タグ「gen」のみが考慮されます。また、変数の名前が module.variable_map とチェックポイント パスのリスト変数の間で一致していることも確認しました。タグ「disc」を持つモジュールの変数マップは次のとおりです。

私は持っている

• テンソルフロー: 1.13.1
• tensorflow_hub: 0​​.4.0
• パイソン: 3.5.2

ご協力いただきありがとうございます

Tensorflow Dataset API を使用して、tensorflow Hub を使用してデータセットを初期化したいと考えています。dataset.map 関数を使用して、テキスト データを埋め込みに変換したいと考えています。私の Tensorflow のバージョンは 1.14 です。

一連の文配列を単語埋め込みに変換する elmo v2 modlule を使用したため、次のコードを使用しました。

[batch_size、max_words_in_batch、embedding_size] のようなテキスト配列を埋め込みたいのですが、次のようなエラー メッセージが表示されました。

期待される結果を得るにはどうすればよいですか?

Tensorflow Hub 画像特徴ベクトル モジュールの 1 つを微調整したいとしましょう。モジュールを微調整するには、次のことを行う必要があるため、問題が発生します。

モジュールがResnet50.

言い換えると、モジュールは、trainableフラグを asTrueに設定してインポートされ、train tag. train tagここで、モデルを検証する (モデルのパフォーマンスを測定するために検証セットで推論を実行する) 場合に備えて、フラグとフラグのためにバッチノルムをオフにすることはできませんtrainable。

この質問はすでにここで尋ねられていることに注意してくださいTensorflow ハブの微調整と評価ですが、回答は提供されていません。

また、それに関する問題についてGithubの問題を提起しました。

あなたの助けを楽しみにしています！

この TF Hub アセットを使用したい: https://tfhub.dev/google/imagenet/resnet_v1_50/feature_vector/3

バージョン:

コード

私は得る：

私は試した：

しかし、モデルを保存しようとすると、次のようになります。

チュートリアルはこちら