Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[fix] DEHvdCheckpoint failed when importing Horovod unsuccessfully. #372

Merged
merged 1 commit into from
Nov 26, 2023

Conversation

MoFHeka
Copy link
Collaborator

@MoFHeka MoFHeka commented Nov 26, 2023

Description

When result of importing Horovod is None, TFRA DEHvdCheckpoint would not call DE variable saving and sweeping redundant DE files.
This problem because I forgot writing the correct code when importing horovod.tensorflow is None in _de_hvd_write_fs_func. Local variable 'de dir' referenced before assignment when self._hvd is None.

Type of change

  • Bug fix
  • New Tutorial
  • Updated or additional documentation
  • Additional Testing
  • New Feature

Checklist:

  • I've properly formatted my code according to the guidelines
    • By running yapf
    • By running clang-format
  • This PR addresses an already submitted issue for TensorFlow Recommenders-Addons
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works

How Has This Been Tested?

Uninstalling Horovod and using DEHvdCheckpoint.

…would not call DE variable saving and sweeping redundant DE files.
@MoFHeka MoFHeka requested a review from rhdong as a code owner November 26, 2023 08:21
Copy link
Member

@rhdong rhdong left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@rhdong rhdong merged commit b078929 into tensorflow:master Nov 26, 2023
40 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants