forked from apache/mxnet
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Infer dtype in SymbolBlock import from input symbol (apache#12412)
* Infer dtype in SymbolBlock import from input symbol * Fix lint issues and make existing tests pass * Add tests for importing a fp64 model into symbol block * Fixing failing test for test symbol block * Set context in unit tests * Add tests for fp16, add default dtype in infer_param_types * Use tmp directory as root for loading from model zoo to avoid race condition * Fixing naming and parameter selection in test case * Fixing failing GPU tests * Make unit test more deterministic to get param name * Override cast in symbol block, handle grouped symbol * Handle multiple symbolic input usecase * Add tests to verify behavior of SymbolBlock.cast
- Loading branch information
1 parent
1744b0c
commit acf309e
Showing
4 changed files
with
151 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters