explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-11 09:00:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Hackishly fix resizing. 3 failures

Browse Source
This commit is contained in:
Matthew Honnibal 2021-10-31 01:28:20 +02:00
parent 931b3e112b
commit 44b01d2a87
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/ml/tb_framework.py
View File

@ -71,7 +71,12 @@ def resize_output(model: Model, new_nO: int) -> Model:
model.attrs["unseen_classes"].add(i)
model.set_param("upper_W", new_W)
model.set_param("upper_b", new_b)
model.set_dim("nO", new_nO, force=True)
# TODO: Avoid this private intrusion
model._dims["nO"] = new_nO
if model.has_grad("upper_W"):
model.set_grad("upper_W", model.get_param("upper_W") * 0)
if model.has_grad("upper_b"):
model.set_grad("upper_b", model.get_param("upper_b") * 0)
return model