Views
No views yet
⚠️ Note: This is the basic version of our ongoing development.
A significantly improved version trained on much larger and more diverse optical corpora will be released soon!
| Parameter | Value | Description |
|---|---|---|
| batch_size | 64 | Number of samples per training batch |
| epochs | 15 | Number of training epochs |
| patience | 6 | Early stopping patience |
| learning_rate | 5e-5 | Learning rate for the AdamW optimizer |
| weight_decay | 0.01 | Weight decay for regularization |
| objective | MLM | Masked Language Modeling |
transformers library by Hugging Face.1from transformers import RobertaTokenizerFast, RobertaModel
2
3tokenizer = RobertaTokenizerFast.from_pretrained("quantum-leap-vcti/VCTI-RoBERTa-Fiber")
4model = RobertaModel.from_pretrained("quantum-leap-vcti/VCTI-RoBERTa-Fiber")
5
6text = "Wavelength-division multiplexing increases the capacity of optical fibers."
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model(**inputs)
9embedding = outputs.last_hidden_state.mean(dim=1)