verl-project/verl

Fused CE loss integration

Ouverte

#97 ouverte le 12 janv. 2025

Voir sur GitHubGuide contributeur
 (3 commentaires) (1 réaction) (0 personne assignée)Python (4 434 forks)auto 404
call for contributionhelp wanted

Métriques du dépôt

Stars
 (23 069 étoiles)
Métriques de merge PR
 (Merge moyen 3j 6h) (125 PRs mergées en 30 j)

Description

Integrate it with main stream models: https://github.com/apple/ml-cross-entropy so that model with large vocab size uses much less memory

Guide contributeur

Direction de recherche
Explorez la fusion de la perte d'entropie croisée de apple/ml cross entropy dans la boucle d'entraînement des modèles principaux de verl project/verl. Identifiez la fonction de perte actuelle du modèle et remplacez la par la version fusionnée, en garantissant le calcul du gradient et les économies de mémoire.
Stack technique
pythonpytorch
Domaine
backendai
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.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Une demi journée
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
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60