-
Notifications
You must be signed in to change notification settings - Fork 463
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.
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
Add FP16 weight and output support #2638
Conversation
This pull request was exported from Phabricator. Differential Revision: D56578193 |
✅ Deploy Preview for pytorch-fbgemm-docs ready!
To edit notification comments on pull requests, go to your Netlify site configuration. |
Summary: This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Differential Revision: D56578193
This pull request was exported from Phabricator. Differential Revision: D56578193 |
Summary: This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Differential Revision: D56578193
Summary: This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Differential Revision: D56578193
Summary: This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Differential Revision: D56578193
Summary: This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Differential Revision: D56578193
This pull request was exported from Phabricator. Differential Revision: D56578193 |
Summary: Pull Request resolved: pytorch#2638 This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Reviewed By: henrylhtsang Differential Revision: D56578193
This pull request was exported from Phabricator. Differential Revision: D56578193 |
Summary: Pull Request resolved: pytorch#2638 This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Reviewed By: henrylhtsang Differential Revision: D56578193
Summary: Pull Request resolved: pytorch#2641 follow up to pytorch#2620 Differential Revision: https://internalfb.com/D57938517
This pull request was exported from Phabricator. Differential Revision: D56578193 |
Summary: Pull Request resolved: pytorch#2638 This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Reviewed By: henrylhtsang Differential Revision: D56578193
Summary: Pull Request resolved: pytorch#2638 This diff adds FP16 weight and output type support in SSD TBE. Changes include: - Add `weights_precision` and `output_dtype` options in `SSDTableBatchedEmbeddingBags` - Update SSD `Initializer` to support FP16 weight - Update SSD TBE unit tests to validate FP16 weight and output Reviewed By: henrylhtsang Differential Revision: D56578193
This pull request was exported from Phabricator. Differential Revision: D56578193 |
This pull request has been merged in 5f9f877. |
Summary:
This diff adds FP16 weight and output type support in SSD TBE.
Changes include:
weights_precision
andoutput_dtype
options inSSDTableBatchedEmbeddingBags
Initializer
to support FP16 weightDifferential Revision: D56578193