Sync trainer state with evaluators
Open
Nobody has claimed this yet.
enhancement
Hacktoberfest
help wanted
needs-discussion
- Dominant language
- Python
- Stars
- 4.8k
- Forks
- 726
- Avg merge
- 5d 21h
- Merged PRs (30d)
- 5
Description
🚀 Feature
There can be use-cases when we would like to get trainer's epoch/iteration or/and other items from trainer.state. Let's propose an API such that we could get easily trainer's state from evaluator.
Context : https://discuss.pytorch.org/t/get-current-epoch-inside-process-function-of-evaluator/162926
Contributor guide
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
Start by reviewing how trainer.state and evaluators currently interact, then examine the linked discussion for the proposed API expectations. Done means evaluators can access the trainer's epoch, iteration, and other relevant state through a clearly defined API.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100