tensorflow / tensorflow/java

API to read weights from ckpt file

Aperta
#420 5 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Java
Stelle
928
Fork
227
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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.

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia confrontando la funzionalità richiesta del Java SDK con CheckPointReader di Python e verificando come vengono rappresentati i pesi dei checkpoint TensorFlow per i client Java e Android. Il lavoro è completato quando viene aggiunta una API Java che legge i pesi dai file ckpt senza modificare l'API preesistente, con un comportamento adatto ai casi d'uso del federated learning.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
java, tensorflow
Ambito
api, machine-learning
Tipo di issue
Funzionalità
Difficoltà
5/5
Tempo stimato
Più di una settimana
Stato di attività
Ferma
Chiarezza
Abbastanza chiara
Idoneità per principianti
25/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.