Utiliser RL-Baselines3-Zoo sur Hugging Face

rl-baselines3-zoo est un framework d'entraînement pour l'apprentissage par renforcement utilisant Stable Baselines3.

Explorer RL-Baselines3-Zoo dans le Hub

Vous pouvez trouver les modèles RL-Baselines3-Zoo en filtrant à gauche de la page des modèles.

L'équipe Stable-Baselines3 héberge une collection de plus de 150 agents d'apprentissage par renforcement entraînés avec des hyperparamètres optimisés, que vous pouvez trouver ici.

Tous les modèles du Hub sont dotés de fonctionnalités utiles :

1. Une carte de modèle générée automatiquement, avec une description, une configuration d'entraînement et plus encore.
2. Balises de métadonnées qui aident à la découvrabilité.
3. Résultats d'évaluation à comparer avec d'autres modèles.
4. Un widget vidéo où vous pouvez voir votre agent en action.

Utilisation de modèles existants

Vous pouvez simplement télécharger un modèle depuis le Hub à l'aide de load_from_hub :

# Download ppo SpaceInvadersNoFrameskip-v4 model and save it into the logs/ folder
python -m rl_zoo3.load_from_hub --algo dqn --env SpaceInvadersNoFrameskip-v4 -f logs/ -orga sb3
python enjoy.py --algo dqn --env SpaceInvadersNoFrameskip-v4  -f logs/

Vous pouvez définir trois paramètres :

• --repo-name : Le nom du dépôt.
• -orga : un nom d'utilisateur ou une organisation Hugging Face.
• -f : le dossier de destination.

Partager vos modèles

Vous pouvez facilement téléverser vos modèles avec push_to_hub. Cela enregistre le modèle, l’évalue, génère une carte de modèle et enregistre une vidéo de replay de votre agent avant de pousser le dépôt complet vers le Hub.

python -m rl_zoo3.push_to_hub  --algo dqn  --env SpaceInvadersNoFrameskip-v4  --repo-name dqn-SpaceInvadersNoFrameskip-v4  -orga ThomasSimonini  -f logs/

Vous pouvez définir trois paramètres :

• --repo-name : Le nom du dépôt.
• -orga : votre nom d'utilisateur Hugging Face.
• -f : le dossier où le modèle est enregistré.

Ressources supplémentaires

• RL-Baselines3-Zoo modèles entraînés officiels
• RL-Baselines3-Zoo documentation