20ca9d882e448df5f1a9c5c0d74772763651826a,thinc/model.py,Model,use_params,#Model#Any#,296

Before Change

        yield
        if backup is not None:
            copy_array(dst=self._mem.weights, src=backup)
        for i, context in enumerate(contexts):
            // This is ridiculous, but apparently it"s what you
            // have to do to make this work across Python 2/3?
            try:
                next(context.gen)
            except StopIteration:
                pass

    def walk(self) -> Iterable["Model"]:
        Iterate out layers of the model, breadth-first.
        queue = [self]
        seen: Set[int] = set()

After Change

            param = params[self.id]
            backup = weights.copy()
            copy_array(dst=weights, src=param)
        with contextlib.ExitStack() as stack:
            for layer in self.layers:
                stack.enter_context(layer.use_params(params))
            for shim in self.shims:
                stack.enter_context(shim.use_params(params))
            yield
        if backup is not None:
            copy_array(dst=self._mem.weights, src=backup)

    def walk(self) -> Iterable["Model"]:

Instances