View all ctranslate2 analysis

How to use the ctranslate2.specs.transformer_spec.TransformerSpec function in ctranslate2

To help you get started, we’ve selected a few ctranslate2 examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

OpenNMT / CTranslate2 / python / ctranslate2 / specs / catalog.py View on Github

"""Catalog of model specifications."""

from ctranslate2.specs import transformer_spec


class TransformerBase(transformer_spec.TransformerSpec):
    def __init__(self):
        super(TransformerBase, self).__init__(6, 8)

class TransformerBig(transformer_spec.TransformerSpec):
    def __init__(self):
        super(TransformerBig, self).__init__(6, 16)