Skip to content

Commit e8e6c58

Browse files
committed
CU-8698ek477: Fix AdamW import from tranformers to torch (#523)
* CU-8698ek477: Add TODO to MetaCAT ML utils regarding AdamW import * CU-8698ek477: Fix AdamW import (trf->torch)
1 parent 09c0ee6 commit e8e6c58

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

medcat/utils/meta_cat/ml_utils.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@
1414
from sklearn.metrics import classification_report, precision_recall_fscore_support, confusion_matrix
1515
from sklearn.model_selection import train_test_split
1616
from sklearn.utils.class_weight import compute_class_weight
17-
from transformers import AdamW, get_linear_schedule_with_warmup
17+
from transformers import get_linear_schedule_with_warmup
18+
from torch.optim import AdamW
1819

1920

2021
import logging

0 commit comments

Comments
 (0)