-
-
Notifications
You must be signed in to change notification settings - Fork 931
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
Conversation
There was a problem hiding this 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.
tests/e2e/multigpu/test_eval.py
Outdated
from transformers.testing_utils import get_torch_dist_unique_port | ||
from utils.tensorboard_ import check_tensorboard |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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
There was a problem hiding this comment.
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.
* replace tensorboard checks with helper function * move helper function * use relative
* replace tensorboard checks with helper function * move helper function * use relative
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.