Instructions to use LieUr/tensorized-bert-rte-archive with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use LieUr/tensorized-bert-rte-archive with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("LieUr/tensorized-bert-rte-archive", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Add archive model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,66 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
library_name: transformers
|
| 4 |
+
tags:
|
| 5 |
+
- bert
|
| 6 |
+
- tensorized
|
| 7 |
+
- pytorch
|
| 8 |
+
- rte
|
| 9 |
+
- research-artifact
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# colaBertFt checkpoint archive
|
| 13 |
+
|
| 14 |
+
This repository is a preservation archive for checkpoints produced by the
|
| 15 |
+
[`colaBertFt`](https://github.com/iAmALai/colaBertFt) research code. The models
|
| 16 |
+
use a BERT-large-shaped architecture with tensorized attention and feed-forward
|
| 17 |
+
layers. One pair uses low-rank activations throughout; the other restricts the
|
| 18 |
+
low-rank activation behavior according to `only_lr_act` in its config.
|
| 19 |
+
|
| 20 |
+
This is an archival release, not a polished inference package. Loading the
|
| 21 |
+
exports may require the custom modeling code in the source repository.
|
| 22 |
+
|
| 23 |
+
## Contents
|
| 24 |
+
|
| 25 |
+
| Remote path | Description |
|
| 26 |
+
|---|---|
|
| 27 |
+
| `bak/chkpt_not_onlylr.pt` | Epoch-2 PyTorch training snapshot with model, optimizer, gradient-scaler, and epoch state. |
|
| 28 |
+
| `bak/chkpt_onlylr.pt` | Epoch-2 `only_lr_act` training snapshot with model, optimizer, gradient-scaler, and epoch state. |
|
| 29 |
+
| `checkpoints/cola_bert_llama_mlp_large_0.7-004/` | FP16 model state dict and architecture config corresponding exactly to the model tensors in `bak/chkpt_not_onlylr.pt`. |
|
| 30 |
+
| `checkpoints/cola_bert_llama_mlp_large_0.7_only_lr-001/` | FP16 model state dict and architecture config corresponding exactly to the model tensors in `bak/chkpt_onlylr.pt`. |
|
| 31 |
+
| `model_checkpoints/rte_cola_bert_llama_mlp_large_0.7_only_lr-001/` | PyTorch Lightning RTE checkpoint from epoch 1/global step 68. Its filename records validation accuracy 0.7329; this archive did not independently reproduce that evaluation. |
|
| 32 |
+
| `model_checkpoints/colaBert_rte_mlpId_0.7329_2e2e91ae/` | Converted FP16 safetensors RTE export, config, and the original generated model card. |
|
| 33 |
+
|
| 34 |
+
Exact SHA-256 checksums are stored in `ARTIFACT_CHECKSUMS.sha256` here and in
|
| 35 |
+
the source repository.
|
| 36 |
+
|
| 37 |
+
## Provenance and training
|
| 38 |
+
|
| 39 |
+
- Base model named by the training configuration: `bert-large-uncased`.
|
| 40 |
+
- Downstream task: GLUE RTE.
|
| 41 |
+
- RTE Lightning checkpoint metadata: 2 labels, 2,490 training samples,
|
| 42 |
+
learning rate 2e-5, Adam epsilon 1e-8, warmup proportion 0.1, weight decay
|
| 43 |
+
0.01, and 117 planned training steps.
|
| 44 |
+
- Archive source files were created in August and September 2025 and archived
|
| 45 |
+
in July 2026.
|
| 46 |
+
|
| 47 |
+
The two 2.4 GB training snapshots contain optimizer state and are retained for
|
| 48 |
+
exact training-resume preservation. Their 615 model tensors match their
|
| 49 |
+
corresponding smaller `model.pt` exports exactly.
|
| 50 |
+
|
| 51 |
+
## Loading and safety
|
| 52 |
+
|
| 53 |
+
The `.pt` and `.ckpt` files use Python/PyTorch serialization. Only load them
|
| 54 |
+
when you trust this archive, and prefer a current PyTorch release with
|
| 55 |
+
`weights_only=True` where compatible. The `.safetensors` export avoids pickle
|
| 56 |
+
deserialization but still requires the matching custom architecture.
|
| 57 |
+
|
| 58 |
+
These artifacts are research outputs and have not been audited for production
|
| 59 |
+
use, bias, robustness, or generalization beyond the recorded RTE experiment.
|
| 60 |
+
|
| 61 |
+
## Licensing
|
| 62 |
+
|
| 63 |
+
The upstream `bert-large-uncased` model is distributed under Apache-2.0. The
|
| 64 |
+
source project does not currently declare a license of its own, so this archive
|
| 65 |
+
does not make an additional licensing grant for the custom code or derived
|
| 66 |
+
artifacts. Users are responsible for checking the applicable upstream terms.
|