Make resizable layer work with textcat and transformers

thinc/layers/resizable.py

@@ -59,6 +59,9 @@ def resize_linear_weighted(
         return layer
     elif new_nO == layer.get_dim("nO"):
         return layer
+    elif layer.has_dim("nI") is None:
+        layer.set_dim("nO", new_nO, force=True)
+        return layer
 
     dims = {name: layer.maybe_get_dim(name) for name in layer.dim_names}
     dims["nO"] = new_nO

thinc/tests/layers/test_resizable.py

@@ -0,0 +1,32 @@
+import pytest
+from functools import partial
+from thinc.api import resizable, Linear
+from thinc.layers.resizable import resize_model, resize_linear_weighted
+
+
+@pytest.fixture
+def model():
+    output_layer = Linear(nO=None, nI=None)
+    fill_defaults = {"b": 0, "W": 0}
+    model = resizable(
+        output_layer,
+        resize_layer=partial(resize_linear_weighted, fill_defaults=fill_defaults),
+    )
+    return model
+
+
+def test_resizable_linear_default_name(model):
+    assert model.name == "resizable(linear)"
+
+
+def test_resize_model(model):
+    """Test that resizing the model doesn't cause an exception."""
+    resize_model(model, new_nO=10)
+    resize_model(model, new_nO=11)
+
+    model.set_dim("nO", 0, force=True)
+    resize_model(model, new_nO=10)
+
+    model.set_dim("nI", 10, force=True)
+    model.set_dim("nO", 0, force=True)
+    resize_model(model, new_nO=10)