pytorch/ignite

Sync trainer state with evaluators

Open

#2 733 ouverte le 6 oct. 2022

Voir sur GitHub
 (10 commentaires) (0 réactions) (0 assignés)Python (4 313 stars) (602 forks)batch import
Hacktoberfestenhancementhelp wantedneeds-discussion

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

Guide contributeur

Stack technique
pythonpytorch
Domaine
machine learning
Type d'issue
feature
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 days
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
stale
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
clear
Prérequis
PyTorch Ignite basicsUnderstanding of trainer and evaluator
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
50
Direction de recherche
Investigate the trainer.state object in ignite/engine/engine.py and the evaluator's process function. Review the discussion at https://discuss.pytorch.org/t/get current epoch inside process function of evaluator/162926 to understand the use case. Propose an API design that allows evaluators to access trainer state, ensuring backward compatibility.