Skip to content

Commit

Permalink
[feat] Make saving model more easier when using HvdAllToAllEmbedding …
Browse files Browse the repository at this point in the history 
…by adding save function overwriting patch in tf_save_restore_patch.py.

Also fix some import bug in tf_save_restore_patch.py.
Also adding a save and restore test for HvdAllToAllEmbeeding.
  • Loading branch information
@MoFHeka
MoFHeka committed Sep 22, 2023
1 parent 0ff7384 commit 9b9f394
Show file tree
Hide file tree
Showing 4 changed files with 629 additions and 83 deletions.
Loading

0 comments on commit 9b9f394

Please sign in to comment.