umassma-terials's picture
Upload 2 files
402bfed verified
|
Raw
History Blame Contribute Delete
816 Bytes
metadata
license: mit
tags:
  - adafactor
  - cnn-transformer
  - concat-mlp
  - grouped-query
  - huge
  - instancenorm
  - mish
  - polynomial
  - retrieval
  - xavier

model_068711920_cnn_transformer_huge.py

Model Overview

A huge-scale implementation of the cnn transformer architecture, built for retrieval tasks.

Architecture

  • Architecture: cnn transformer
  • Scale: huge
  • Attention: grouped query
  • Fusion strategy: concat mlp
  • Task head: retrieval
  • Activation: mish
  • Normalization: instancenorm
  • Initialization: xavier

Training

  • Optimizer: adafactor
  • LR scheduler: polynomial

Files

  • model_068711920_cnn_transformer_huge.py — main artifact of this repository

License

See the license field above.