예제 #1
0
 def __init__(self, upstream_name: str):
     Model.__init__(self,
                    name=self.name,
                    forward=forward,
                    dims={"nO": None})
     self.upstream_name = upstream_name
     self._batch_id = None
     self._outputs = None
     self._backprop = None
예제 #2
0
 def __init__(self, upstream_name: str, width: int) -> None:
     """
     upstream_name (str): A string to identify the 'upstream' Tok2Vec component
         to communicate with. The upstream name should either be the wildcard
         string '*', or the name of the `Tok2Vec` component. You'll almost
         never have multiple upstream Tok2Vec components, so the wildcard
         string will almost always be fine.
     width (int):
         The width of the vectors produced by the upstream tok2vec component.
     """
     Model.__init__(self, name=self.name, forward=forward, dims={"nO": width})
     self.upstream_name = upstream_name
     self._batch_id = None
     self._outputs = None
     self._backprop = None