lora-retriever / README.md
codysyun's picture
Upload 2 files
02b3314 verified
|
Raw History Blame Contribute Delete
698 Bytes
metadata
license: cc-by-4.0
tags:
  - base
  - hybrid
  - kaiming
  - lamb
  - matching
  - onecycle
  - relu
  - scalenorm
  - standard
  - tucker

predict.py

Model Overview

A base-scale implementation of the hybrid architecture, built for matching tasks.

Architecture

  • Architecture: hybrid
  • Scale: base
  • Attention: standard
  • Fusion strategy: tucker
  • Task head: matching
  • Activation: relu
  • Normalization: scalenorm
  • Initialization: kaiming

Training

  • Optimizer: lamb
  • LR scheduler: onecycle

Files

  • predict.py — main artifact of this repository

License

See the license field above.