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
161 stars 83 forks source link

fix: fix tf2 test for tf 2.11 release #626

Closed yl-to closed 2 years ago

yl-to commented 2 years ago

Description of changes:

from tf 2.11, the keras optimizers has been reworked and pointed to optimizer_experimental versions: https://github.com/keras-team/keras/tree/v2.11.0-rc3/keras/optimizers/optimizer_experimental This caused a failing test in tf 2.11 DLC release. In this PR, we replaced the optimizer in test file with the previously used v2 adam optimizer to help unblock tf 2.11 release.

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.