From b6de8cf465b2b932880c6cf8bb75b9ee9cfe1849 Mon Sep 17 00:00:00 2001 From: JacJacJr Date: Tue, 15 Apr 2025 21:44:43 +0200 Subject: [PATCH] Update import of AdamW optimizer - curent from tourch.optim, not transformer anymore --- beir/retrieval/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/beir/retrieval/train.py b/beir/retrieval/train.py index f80f9f15..4592b17b 100644 --- a/beir/retrieval/train.py +++ b/beir/retrieval/train.py @@ -14,10 +14,9 @@ ) from sentence_transformers.readers import InputExample from torch import nn -from torch.optim import Optimizer +from torch.optim import Optimizer, AdamW from torch.utils.data import DataLoader from tqdm.autonotebook import trange -from transformers import AdamW logger = logging.getLogger(__name__)