EMOTYC ONNX Models
Ce dépôt contient deux conversions ONNX de modèles EMOTYC pour l'identification d'émotions dans des textes écrits en français.
Fichiers
EMOTYC_1/model.onnx: conversion ONNX de la versionsafetensorspubliée sur Hugging Face.EMOTYC_2/model.onnx: conversion ONNX d'une version ultérieure issue d'un checkpoint PyTorch présent dans le dépôt GitLab TextToKids.config.jsonettokenizer.json: configuration et tokenizer communs utilisés avec les deux exports ONNX.
Origine des modèles
EMOTYC_1
EMOTYC_1 correspond à une conversion ONNX du modèle EMOTYC publié ici :
https://huggingface.co/TextToKids/CamemBERT-base-EmoTextToKids/blob/main/model.safetensors
Cette première version est associée au modèle décrit dans l'article suivant :
https://arxiv.org/abs/2405.14385
L'article documente le fine-tuning complet de CamemBERT-base pour obtenir un modèle de classification multi-label destiné à prédire si une phrase exprime une émotion, ses modes d'expression, son caractère basique ou complexe, et sa catégorie émotionnelle.
EMOTYC_2
EMOTYC_2 correspond à une conversion ONNX de cette version du modèle EMOTYC :
Le checkpoint source se trouve dans le dossier GitLab suivant :
Ce dossier contient aussi des fichiers d'entraînement comme trainer_state.json, optimizer.pt, rng_state.pth, scheduler.pt et training_args.bin. Ces fichiers peuvent aider à comprendre l'état du checkpoint, mais ils ne sont pas inclus dans ce dépôt ONNX.
Hypothèse de filiation
La version safetensors semble précéder la version PyTorch du checkpoint GitLab. La pipeline probable est donc la suivante :
- départ de
CamemBERT-base; - fine-tuning complet documenté dans l'article arXiv, produisant
EMOTYC_1; - nouveau fine-tuning ou poursuite d'entraînement produisant
EMOTYC_2.
Le passage de CamemBERT-base à EMOTYC_1 est documenté par l'article arXiv. En revanche, la manière exacte dont EMOTYC_2 a été obtenu n'est pas complètement documentée publiquement, même si le dossier GitLab du checkpoint fournit plusieurs artefacts d'entraînement.
Notes
Les deux fichiers ONNX ont été conservés séparément car leurs contenus binaires diffèrent. Les fichiers PyTorch et les états d'entraînement du checkpoint source n'ont pas été poussés dans ce dépôt.
- Downloads last month
- 3