Lightning-AI/pytorch-lightning

Resuming should allow to differentiate what to resume (steps/opti/weights)

Open

#5 339 ouverte le 3 janv. 2021

Voir sur GitHub
 (26 commentaires) (18 réactions) (0 assignés)Python (3 233 forks)batch import
featurehelp wantedpriority: 1

Métriques du dépôt

Stars
 (26 687 stars)
Métriques de merge PR
 (Merge moyen 9j 15h) (3 PRs mergées en 30 j)

Description

Currently it is possible to either resume only the full training state (epoch/global steps / optimizer / scheduler options / and weights), or only the weights.

I would like to be able to switch the optimizer at some point, i.e. skip restoring optimizer/scheduler, but still load the epoch/global steps. I only see a way to do this with hacks at the moment. Any other way? Could this be a feature, to specify in the trainer.init function specifically what to restore/what not to restore?

Guide contributeur

Direction de recherche
Examinez la logique actuelle de restauration des points de contrôle dans Lightning et modifiez le `Trainer` ou `CheckpointConnector` pour accepter un paramètre permettant de restaurer sélectivement les composants de l'état d'entraînement.
Stack technique
python
Domaine
backend
Type d'issue
Fonctionnalité
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-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch Lightning
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
40