awslabs / sagemaker-debugger

Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors
Apache License 2.0
158 stars 82 forks source link

Fix dt uint64 missing proto #617

Closed dkey-amazon closed 1 year ago

dkey-amazon commented 1 year ago

Description of changes:

Adding a few extra types for the SMDebug Tensorflow protobufs. Without these types, I hit some failing tests trying to use SMDebug for TF 2.10. With these types, the tests pass in my test environment.

See test dlc-pr-tensorflow-2_dlc-pr-ec2-test at https://github.com/aws/deep-learning-containers/pull/2181 for failure details.

Also, in order to invalidate the previous 1.0.20 tag, I'm bumping the SMDebug version to 1.0.21.

Style and formatting:

I have run pre-commit install && pre-commit run --all-files to ensure that auto-formatting happens with every commit.

Issue number, if available

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.