Spaces:

CadenShokat
/

modernbert-finetuned-embeddings

Paused

Caden Shokat commited on Aug 15

Commit

13f746f

1 Parent(s): 71adeda

bf16 false

Files changed (1) hide show

src/training/train.py CHANGED Viewed

@@ -44,7 +44,7 @@ def main():
         lr_scheduler_type="cosine",
         optim="adamw_torch_fused",
         tf32=False,
-        bf16=True,
         batch_sampler=BatchSamplers.NO_DUPLICATES,
         eval_strategy="epoch",
         save_strategy="epoch",

         lr_scheduler_type="cosine",
         optim="adamw_torch_fused",
         tf32=False,
+        bf16=False,
         batch_sampler=BatchSamplers.NO_DUPLICATES,
         eval_strategy="epoch",
         save_strategy="epoch",