Model Details

google/gemma-2b model finetuned on 100,000 CLRS-Text examples.

Training Details

  • Learning Rate: 1e-4, 150 warmup steps then cosine decayed to 5e-06 using AdamW optimiser
  • Batch size: 128
  • Loss taken over answer only, not on question.
Downloads last month
171
Safetensors
Model size
2.51B params
Tensor type
BF16
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.

Model tree for smcleish/clrs_gemma_2b_100k_finetune_with_traces

Base model

google/gemma-2b
Finetuned
(195)
this model