How to attach the apex logger to my training code
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 9k
- Forks
- 1.5k
- Avg merge
- 2d 4h
- Merged PRs (30d)
- 3
Description
This is probably more of a basic python question, but how do I attach the apex.amp logger to the logger used in my training code?
These types of messages are showing up in the console, but are not getting caught by my logger:
Gradient overflow. Skipping step, loss scaler 0 reducing loss scale to 16384.0
Thanks!
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
No file or test is named in the issue. Start by locating the apex.amp logger output and comparing it with Python's logger used by the training code; done means the gradient-overflow messages are captured by that logger or the required integration is documented.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning, tooling
- Issue type
- Documentation
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100