tensorflow / tensorflow/java

API to read weights from ckpt file

オープン
#420 コメント 5 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

主要言語
Java
スター
928
フォーク
227
PR マージ指標
30日以内にマージされた PR はありません

説明

System information

  • TensorFlow version (you are using): Nightly
  • Are you willing to contribute it (Yes/No): No

Describe the feature and the current behavior/state.
This API would help in reading the weights from ckpt file. Currently, there is no such feature in the Java SDK but the same can be achieved in python via CheckPointReader.

Will this change the current API? How?
This shouldn't affect the preexisting API. It would be a simple addition over it.

Who will benefit with this feature?
I was developing a Federated Learning Application and was stuck at the issue where I was not able to get the weights from the ckpt file (Reason: No API in Java SDK) and neither could I write a signature method in tflite for the same (Reason: tflite doesn't support dynamic tensor shapes). So this would help promote active development in FL on Android clients.

コントリビューションガイド

コントリビューションガイドを開く

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

調査の方向性

まず、要求されている Java SDK の機能を Python の CheckPointReader と比較し、TensorFlow のチェックポイントの重みが Java および Android クライアント向けにどのように表現されているかを確認します。完了条件は、既存の API を変更せずに ckpt ファイルから重みを読み取る Java API を追加し、連合学習のユースケースに適した動作にすることです。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
java, tensorflow
領域
api, machine-learning
issue の種類
機能追加
難易度
5/5
見積もり時間
1週間以上
活発さ
停滞
明瞭さ
おおむね明確
初心者へのやさしさ
25/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。