How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
# Warning: Pipeline type "translation" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline

pipe = pipeline("translation", model="rdj-034/lab2_efficient")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("rdj-034/lab2_efficient")
model = AutoModelForSeq2SeqLM.from_pretrained("rdj-034/lab2_efficient")
Quick Links

lab2_efficient

Hyperparameters

  • learning_rate: 2e-5
  • per_device_train_batch_size: 128
  • effective_batch_size: 128
  • gradient_accumulation_steps: 1
  • weight_decay: 0.1
  • optimizer: adamw_torch
  • fp16: True
  • gradient_checkpointing: True
  • lr_scheduler: cosine
  • warmup_ratio: 0.1
  • max_steps: 100

Results

Metric Value
BLEU 44.113
Eval Loss 1.3546
Train Steps 100
Epoch 0.0615
Downloads last month
7
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Evaluation results