-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
Migrate underlying framework from TensorFlow to PyTorch #1266
Merged
Conversation
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
* refactor: correct data type for specifying input_size * refactor: replace tf apis w/ torch, add input_size specification * refactor: tensor manipulation and testing for expected shape * refactor: partial working with attention reducer * code: clean out tf commented out code * debug: placehoder code * refactor: use proper Torch container for dependency_reducers * refactor: back-off special handling of attention reducer This will need to revaluated becuase this reducer has parameters that need special handling to support attention reducer initialization.
* Updated GitHub actions * Fixed check * Updated AverageMeter to MeanMetric * Lower bound torchmetrics
…assing test_experiment. (#1443) * WIP for test_experiment. * Fixes to StackedTransformer, image_utils, sequence_feature, and how images are read in test_experiment to address all test failures. * Minor cleanup and test parameterization.
* refactor: removed deprecated sequence encoder test mark decoder related test to be reworked at later date. * test: working version of sequence input feature test
* Remove keras/tf references from test_collect integration tests and detach tensors before serializing to numpy (required by torch). * Fix (and simplify) implementation for collect_weights. * Cleanup. * Fix _get_layers() helper. * Use torch.allclose() to compare tensor values. * Simplify test_collect_activations. * Use named children instead of layers. * Remove import logging
- Removed per module regularization - Added regularization at the ECD-model level - Updated regularization tests to check for end-to-end regularization.
…uropod, and savedmodel (#1452)
- Updated MeanMetric to return a tensor of dimension torch.Size([])
…raining as an additional parameter everywhere. (#1459)
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Continuation of #1155.