sy368vi/Bakalar_work_final
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update train_ner.py

Browse Source
This commit is contained in:
Serhii Yemets 2025-05-28 21:51:05 +00:00
parent be46e325c8
commit 764e49894d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_ner.py
View File

@ -191,6 +191,6 @@ try:
full_model_path = os.path.join(output_base, "full_model") full_model_path = os.path.join(output_base, "full_model")
merged_model.save_pretrained(full_model_path) merged_model.save_pretrained(full_model_path)
tokenizer.save_pretrained(full_model_path) tokenizer.save_pretrained(full_model_path)
print(f"✅ Full merged model saved to {full_model_path}") print(f"Full merged model saved to {full_model_path}")
except Exception as e: except Exception as e:
print(f"❌ Failed to merge and save full model: {e}") print(f"Failed to merge and save full model: {e}")