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

replace tensorboard checks with helper function #2120

Merged
merged 3 commits into from
Dec 4, 2024
Merged

Conversation

winglian
Copy link
Collaborator

@winglian winglian commented Dec 3, 2024

Description

There's a lot of duplicate code and boilerplate for checking tensorboard log values. This helper function should make it much easier to check values from e2e trainings without a lot of extra code.

Copy link
Contributor

@djsaunde djsaunde left a comment

Choose a reason for hiding this comment

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

Seems like an easy win, just a nit about the isort order of the utils module.

from transformers.testing_utils import get_torch_dist_unique_port
from utils.tensorboard_ import check_tensorboard
Copy link
Contributor

Choose a reason for hiding this comment

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

Why is utils sorted as a 3rd party package?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

ah, conflict in utils under axolotl and utils under tests maybe? let me see if I can make that work properly

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

moving the helper fn and converting to a relative import seems to make it more obvious now.

@winglian winglian merged commit a1790f2 into main Dec 4, 2024
16 checks passed
@winglian winglian deleted the test-helper-tbparse branch December 4, 2024 02:06
djsaunde pushed a commit that referenced this pull request Dec 16, 2024
* replace tensorboard checks with helper function

* move helper function

* use relative
djsaunde pushed a commit that referenced this pull request Dec 17, 2024
* replace tensorboard checks with helper function

* move helper function

* use relative
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