File: __init__.py

package info (click to toggle)
pytorch-text 0.14.1-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 11,560 kB
  • sloc: python: 14,197; cpp: 2,404; sh: 214; makefile: 20
file content (19 lines) | stat: -rw-r--r-- 446 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
 
from .bundler import (
    ROBERTA_BASE_ENCODER,
    ROBERTA_LARGE_ENCODER,
    RobertaBundle,
    XLMR_BASE_ENCODER,
    XLMR_LARGE_ENCODER,
)
from .model import RobertaClassificationHead, RobertaEncoderConf, RobertaModel

__all__ = [
    "RobertaEncoderConf",
    "RobertaClassificationHead",
    "RobertaModel",
    "RobertaBundle",
    "XLMR_BASE_ENCODER",
    "XLMR_LARGE_ENCODER",
    "ROBERTA_BASE_ENCODER",
    "ROBERTA_LARGE_ENCODER",
]