micro zapusk no cross

1a49ab7b · Мазур Грета Евгеньевна · 1fbe726d · 1a49ab7b · 1a49ab7b
Commit 1a49ab7b authored 2 weeks ago by Мазур Грета Евгеньевна
Hide whitespace changes
Inline Side-by-side

Showing

with 6 additions and 0 deletions
+6 -0
--- a/.ipynb_checkpoints/checkLora-checkpoint.py
+++ b/.ipynb_checkpoints/checkLora-checkpoint.py
@@ -314,9 +314,12 @@ tokenizer = BertTokenizer.from_pretrained("./micro_no_cross_fine_tuned2")


 base_model = MultiTaskBert.from_pretrained('bert-base-uncased').to(device)
+print("Загружена базовая")

 model = PeftModel.from_pretrained(base_model, "./micro_no_cross_fine_tuned/lora2", strict=False)

+print("Загружена с лора")
+
 # Переводим модель в режим оценки
 model.eval()


--- a/checkLora.py
+++ b/checkLora.py
@@ -314,9 +314,12 @@ tokenizer = BertTokenizer.from_pretrained("./micro_no_cross_fine_tuned2")


 base_model = MultiTaskBert.from_pretrained('bert-base-uncased').to(device)
+print("Загружена базовая")

 model = PeftModel.from_pretrained(base_model, "./micro_no_cross_fine_tuned/lora2", strict=False)

+print("Загружена с лора")
+
 # Переводим модель в режим оценки
 model.eval()