sagemaker-debugger
sagemaker-debugger copied to clipboard
Optimizer variables, Layer inputs/outputs, model inputs/output are not being saved when save_all=True
Noticed this for TF2.
helper_keras_gradtape
and helper_keras_fit
in tensorflow2/test_keras.py