Optimizer variables, Layer inputs/outputs, model inputs/output are not being saved when save_all=True

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

Open rahul003 opened 4 years ago

rahul003 commented 4 years ago

Noticed this for TF2. helper_keras_gradtape and helper_keras_fit in tensorflow2/test_keras.py