Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[js/web/training] added parameters methods implementations #18086

Closed
wants to merge 23 commits into from
Closed

[js/web/training] added parameters methods implementations #18086

wants to merge 23 commits into from

Conversation

carzh
Copy link
Contributor

@carzh carzh commented Oct 24, 2023

Description

  • Added the following methods & their implementations: getParametersSize, loadParametersBuffer, and getContiguousBuffer

Motivation and Context

  • adding training bindings for web
  • these methods are necessary for federated learning, where the workflow includes a model being sent to an edge device, the edge device performing training, then sending the updated weights back to the central server.

Related work

carzh and others added 22 commits October 2, 2023 14:54
@carzh
added training build configuration
cff6a7f
@carzh
edited wasm factory + added training backend
7a1365d
@carzh
added package.json modification for training + minimized training art…
71f9dbc 
…ifact
@carzh
applied suggestions
5adbcf4
@carzh
create training session implementation + supporting changes
9068ab4
@fs-eire
Merge remote-tracking branch 'origin/main' into carzh/training-wasm-b…
edbe9cc 
…inding
@carzh
fixed variable names + enforced wasmBackend being a singleton with su…
5202e3b 
…ggested fix of adding backend-wasm-inference.ts
@carzh
format + lint
a277347
@carzh
Merge branch 'main' into carzh/training-wasm-binding
adfa5cf
@carzh
Merge branch 'carzh/training-wasm-binding' into carzh/create-training…
c9931ad 
…-session
@carzh
minor tweak to remove placeholder return statement
e775933
@carzh
format
c745a65
@carzh
lint fixes
beca6dc
@carzh
lint + format
b37e77f
@carzh
Merge branch 'main' into carzh/create-training-session
8c1e227
@carzh
added isOrtEnvInitialized case to proxy wrapper
d2f3d88
@carzh
Merge remote-tracking branch 'refs/remotes/origin/carzh/create-traini…
712e73c 
…ng-session' into carzh/create-training-session
@carzh
fixed proxy wrapper case statement
05a708f 
fixed proxy worker

run format
@carzh
working runTrainStep implementation
46a9677
@carzh
light refactoring
3ca1c27 
renamed session-handler for inference files

lint + format
@carzh
getContiguousParameters & loadParametersBuffer impl
b44b705 
wrote untested getContiguousParameters method

updated getInputOutputCount and getInputOutputNames signature, added more informative error message

updated parameter names according to suggestions

semi working getContiguousParameters impl

working getContiguousParams, started writing loadParametersBuffer

working version of loadParametersBuffer
@carzh
lint + format + added error code checking wrapper
c74112e
@carzh carzh requested review from askhade and guschmue October 24, 2023 23:10
@carzh carzh mentioned this pull request Oct 26, 2023
Closed
@carzh carzh closed this Nov 2, 2023
@carzh carzh deleted the carzh/web-parameters-methods branch December 20, 2023 22:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@askhade askhade Awaiting requested review from askhade

@guschmue guschmue Awaiting requested review from guschmue

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@carzh @fs-eire