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 hook load times for tf2 #559

Closed mariumof closed 2 years ago

mariumof commented 2 years ago

Description of changes:

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.

codecov-commenter commented 2 years ago

Codecov Report

Merging #559 (e2c1f76) into master (71866e2) will decrease coverage by 1.28%. The diff coverage is n/a.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #559      +/-   ##
==========================================
- Coverage   72.50%   71.21%   -1.29%     
==========================================
  Files         127      117      -10     
  Lines       11171    10693     -478     
==========================================
- Hits         8099     7615     -484     
- Misses       3072     3078       +6     
Impacted Files Coverage Δ
smdebug/tensorflow/reductions.py 0.00% <0.00%> (-47.06%) :arrow_down:
...mdebug/profiler/analysis/notebook_utils/heatmap.py 0.00% <0.00%> (-13.01%) :arrow_down:
smdebug/xgboost/hook.py 0.00% <0.00%> (-8.34%) :arrow_down:
smdebug/core/writer.py 86.20% <0.00%> (-6.04%) :arrow_down:
smdebug/core/logger.py 70.83% <0.00%> (-5.56%) :arrow_down:
smdebug/profiler/utils.py 72.22% <0.00%> (-4.94%) :arrow_down:
smdebug/core/tfevent/index_file_writer.py 93.75% <0.00%> (-1.57%) :arrow_down:
smdebug/profiler/analysis/python_stats_reader.py 94.28% <0.00%> (-1.43%) :arrow_down:
smdebug/tensorflow/collection.py 94.84% <0.00%> (-1.04%) :arrow_down:
smdebug/pytorch/hook.py 75.46% <0.00%> (-0.63%) :arrow_down:
... and 14 more

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 71866e2...e2c1f76. Read the comment docs.