onnxmodelzoo
/

switchtransformersencoder_Opset16

+author: transformers
+builds:
+  x86_ort:
+    all_build_stages:
+    - export_pytorch
+    - set_success
+    benchmark_status: successful
+    completed_build_stages:
+      export_pytorch: 18.960886001586914
+      set_success: 0.0334930419921875
+    device_type: x86
+    iterations: 100
+    onnx_file: /home/azureuser/.cache/turnkey/switchtransformersencoder_transformers_561f22b6/onnx/switchtransformersencoder_transformers_561f22b6-op16-base.onnx
+    runtime: ort
+builtin_model_script: https://github.com/onnx/turnkeyml/blob/main/models/transformers/switchtransformersencoder.py
+class: SwitchTransformersEncoderModel
+hash: 3ac88646
+model_name: switchtransformersencoder
+onnx_input_dimensions:
+  attention_mask:
+  - 1
+  - 128
+  input_ids:
+  - 1
+  - 128
+onnx_model_information:
+  ir_version: 8
+  opset: 16
+  size on disk (KiB): 1203258.499
+onnx_ops_counter:
+  Abs: 1
+  Add: 64
+  ArgMax: 12
+  Cast: 112
+  Constant: 464
+  CumSum: 6
+  Div: 27
+  Expand: 48
+  Gather: 98
+  GatherND: 48
+  Less: 1
+  LessOrEqual: 6
+  Log: 1
+  MatMul: 186
+  Min: 1
+  Mul: 64
+  NonZero: 96
+  OneHot: 6
+  Pow: 25
+  ReduceMax: 6
+  ReduceMean: 25
+  Relu: 54
+  Reshape: 97
+  ScatterND: 48
+  Shape: 96
+  Slice: 48
+  Softmax: 18
+  Sqrt: 25
+  Sub: 1
+  Transpose: 145
+  Unsqueeze: 105
+  Where: 1
+parameters: 307846272
+system_info:
+  Memory Info: 62.79 GB
+  OEM System: Virtual Machine
+  OS Version: Linux-5.15.0-1051-azure-x86_64-with-glibc2.17
+  Processor: Intel(R) Xeon(R) Platinum 8272CL CPU @ 2.60GHz
+  Python Packages:
+  - gitpython==3.1.40
+  - jinja2==3.1.2
+  - markdown==3.5.1
+  - markupsafe==2.1.3
+  - pillow==10.1.0
+  - pynacl==1.5.0
+  - pyyaml==6.0.1
+  - pygments==2.16.1
+  - absl-py==2.0.0
+  - aiohttp==3.8.6
+  - aiosignal==1.3.1
+  - altair==5.1.2
+  - appdirs==1.4.4
+  - async-timeout==4.0.3
+  - attrs==23.1.0
+  - backports.zoneinfo==0.2.1
+  - bcrypt==4.0.1
+  - blinker==1.7.0
+  - boto3==1.28.77
+  - botocore==1.31.77
+  - cachetools==5.3.2
+  - certifi==2023.7.22
+  - cffi==1.16.0
+  - charset-normalizer==3.3.2
+  - click==8.1.7
+  - coloredlogs==15.0.1
+  - cryptography==41.0.5
+  - datasets==2.14.6
+  - diffusers==0.12.0
+  - dill==0.3.7
+  - docker-pycreds==0.4.0
+  - fasteners==0.19
+  - filelock==3.13.1
+  - flatbuffers==23.5.26
+  - frozenlist==1.4.0
+  - fsspec==2023.10.0
+  - fvcore==0.1.5.post20221221
+  - gitdb==4.0.11
+  - google-auth==2.23.4
+  - google-auth-oauthlib==1.0.0
+  - grpcio==1.59.2
+  - huggingface-hub==0.17.3
+  - humanfriendly==10.0
+  - hummingbird-ml==0.4.4
+  - idna==3.4
+  - importlib-metadata==6.8.0
+  - importlib-resources==6.1.0
+  - invoke==2.2.0
+  - iopath==0.1.10
+  - jmespath==1.0.1
+  - joblib==1.3.2
+  - jsonschema==4.19.2
+  - jsonschema-specifications==2023.7.1
+  - lightgbm==3.3.5
+  - markdown-it-py==3.0.0
+  - mdurl==0.1.2
+  - mpmath==1.3.0
+  - multidict==6.0.4
+  - multiprocess==0.70.15
+  - networkx==3.1
+  - nltk==3.8.1
+  - numpy==1.24.4
+  - nvidia-cublas-cu12==12.1.3.1
+  - nvidia-cuda-cupti-cu12==12.1.105
+  - nvidia-cuda-nvrtc-cu12==12.1.105
+  - nvidia-cuda-runtime-cu12==12.1.105
+  - nvidia-cudnn-cu12==8.9.2.26
+  - nvidia-cufft-cu12==11.0.2.54
+  - nvidia-curand-cu12==10.3.2.106
+  - nvidia-cusolver-cu12==11.4.5.107
+  - nvidia-cusparse-cu12==12.1.0.106
+  - nvidia-nccl-cu12==2.18.1
+  - nvidia-nvjitlink-cu12==12.3.52
+  - nvidia-nvtx-cu12==12.1.105
+  - oauthlib==3.2.2
+  - onnx==1.15.0
+  - onnxconverter-common==1.14.0
+  - onnxmltools==1.10.0
+  - onnxruntime==1.15.1
+  - packaging==23.2
+  - pandas==2.0.3
+  - paramiko==2.11.0
+  - pathtools==0.1.2
+  - pip==23.3
+  - pkgutil-resolve-name==1.3.10
+  - portalocker==2.8.2
+  - protobuf==3.20.2
+  - psutil==5.9.6
+  - pyarrow==14.0.0
+  - pyasn1==0.5.0
+  - pyasn1-modules==0.3.0
+  - pycparser==2.21
+  - pydeck==0.8.1b0
+  - pyparsing==3.1.1
+  - python-dateutil==2.8.2
+  - pytorch-pretrained-biggan==0.1.1
+  - pytz==2023.3.post1
+  - referencing==0.30.2
+  - regex==2023.10.3
+  - requests==2.31.0
+  - requests-oauthlib==1.3.1
+  - rich==13.6.0
+  - rjieba==0.1.11
+  - roformer==0.4.3
+  - rpds-py==0.10.6
+  - rsa==4.9
+  - s3transfer==0.7.0
+  - safetensors==0.4.0
+  - scikit-learn==1.1.1
+  - scipy==1.10.1
+  - sentence-transformers==2.2.2
+  - sentencepiece==0.1.99
+  - sentry-sdk==1.34.0
+  - seqeval==1.2.2
+  - setproctitle==1.3.3
+  - setuptools==68.0.0
+  - simpletransformers==0.64.3
+  - six==1.16.0
+  - skl2onnx==1.15.0
+  - smmap==5.0.1
+  - streamlit==1.28.1
+  - sympy==1.12
+  - tabulate==0.9.0
+  - tenacity==8.2.3
+  - tensorboard==2.14.0
+  - tensorboard-data-server==0.7.2
+  - termcolor==2.3.0
+  - threadpoolctl==3.2.0
+  - timm==0.9.8
+  - tokenizers==0.14.1
+  - toml==0.10.2
+  - toolz==0.12.0
+  - torch==2.1.0
+  - torch-geometric==2.4.0
+  - torchaudio==2.1.0
+  - torchvision==0.16.0
+  - tornado==6.3.3
+  - tqdm==4.66.1
+  - transformers==4.35.0
+  - triton==2.1.0
+  - turnkeyml==0.2.0
+  - typeguard==4.1.5
+  - typing-extensions==4.8.0
+  - tzdata==2023.3
+  - tzlocal==5.2
+  - urllib3==1.26.18
+  - validators==0.22.0
+  - wandb==0.15.12
+  - watchdog==3.0.0
+  - werkzeug==3.0.1
+  - wheel==0.41.2
+  - xgboost==1.6.1
+  - xxhash==3.4.1
+  - yacs==0.1.8
+  - yarl==1.9.2
+  - zipp==3.17.0
+task: Natural_Language_Processing