Partager via

accel_model Module

  • Référence

Module avec classe de base abstraite de modèles accélérés matériellement.

Classes

AccelModel

Classe de base abstraite pour les modèles accélérés.

Les modèles accélérés sont des réseaux neuronaux qui peuvent être accélérés à l’aide d’un matériel dédié.

Classe de base abstraite pour un modèle anaccel.

Pour ajouter un nouveau modèle, implémentez une sous-classe : folder_name de modèle, version, checkpoint_uri et save_name doivent tous être donnés par l’utilisateur. Le chemin de poids et est figé doivent être exposés au moins pour les versions quantifiées. :p aram model_base_path : chemin d’accès de base dans lequel stocker tous les modèles. Généralement donné par l’utilisateur. :p aram model_folder_name : chemin d’accès sur disque pour stocker toutes les versions de ce modèle. :p aram version : version de ce modèle. :p aram check_point_uri : URI à partir duquel le modèle est téléchargé s’il n’est pas sur le disque. :p aram save_name : nom sous lequel le point de contrôle est enregistré, utilisé pour charger le métagraphe. :p aram is_frozen : si le modèle doit être figé lors du chargement. Cela fige le graphe en supprimant les variables de tf. GraphKeys.TRAINABLE_VARIABLES. :p aram weight_path : chemin d’accès personnalisé à partir duquel charger les pondérations, au lieu du chemin par défaut sur le disque. Utilisé dans les scénarios de réentraînement.