Revert Pytorch changes for 0.9.4

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

Closed NihalHarish closed 4 years ago

NihalHarish commented 4 years ago

Reverting code changes to skip the test for pytorch 1.7 since the underlying dataloader bug has been fixed.

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

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